fix: pickfile path of upload_worker in CLI mode

biothings · May 17, 2023 · 0157098 · 0157098
1 parent 2acf8ed
commit 0157098
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/biothings/utils/workers.py b/biothings/utils/workers.py
@@ -34,7 +34,8 @@ def upload_worker(name, storage_class, loaddata_func, col_name, batch_size, batc
         )
         import pickle
 
-        pickfile = os.path.join(os.path.dirname(logfile), "%s.pick" % logger_name)
+        # when logfile is None, we're in CLI mode, so we just use the current directory
+        pickfile = os.path.join(os.path.dirname(logfile) if logfile else ".", "%s.pick" % logger_name)
         try:
             pickle.dump(
                 {