Update BERTopic/my_topic_modeling.py
Browse files
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"))
|