fix

chenyangkang · Oct 27, 2024 · d38ff40 · d38ff40
1 parent d0c4b0e
commit d38ff40
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/stemflow/model/AdaSTEM.py b/stemflow/model/AdaSTEM.py
@@ -1263,9 +1263,10 @@ def save(self, tar_gz_file, remove_temporary_file = True):
 
         #
         path, basename = os.path.split(Path(tar_gz_file.rstrip('/\\')))
+        tmp_code = int(np.random.uniform(0,1e8))
 
         # temporary save the model using pickle
-        model_path = os.path.join(path, 'model.pkl')
+        model_path = os.path.join(path, f'model_{tmp_code}.pkl')
         with open(model_path, 'wb') as f:
             pickle.dump(self, f)