kisejin commited on
Commit
e4ff17f
·
verified ·
1 Parent(s): 6199054

Update BERTopic/my_topic_modeling.py

Browse files
Files changed (1) hide show
  1. BERTopic/my_topic_modeling.py +8 -0
BERTopic/my_topic_modeling.py CHANGED
@@ -56,6 +56,14 @@ import torch
56
  from GPUtil import showUtilization as gpu_usage
57
  from numba import cuda
58
 
 
 
 
 
 
 
 
 
59
 
60
  # Get working directory
61
  working_dir = os.path.abspath(os.path.join("/workspace", "TopicModelingRepo"))
 
56
  from GPUtil import showUtilization as gpu_usage
57
  from numba import cuda
58
 
59
+ import pretty_errors
60
+ import datetime
61
+
62
+ pretty_errors.configure(
63
+ display_timestamp=1,
64
+ timestamp_function=lambda: datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S')
65
+ )
66
+
67
 
68
  # Get working directory
69
  working_dir = os.path.abspath(os.path.join("/workspace", "TopicModelingRepo"))