Skip to content

Commit 892d736

Browse files
committed
fix bug
1 parent 567cb67 commit 892d736

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

stemflow/model/AdaSTEM.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1494,7 +1494,7 @@ def load(tar_gz_file, new_lazy_loading_path=None, remove_original_file=False):
14941494
if model.lazy_loading:
14951495
for model_name in model.model_dict:
14961496
if isinstance(model.model_dict[model_name], LazyLoadingEstimator):
1497-
model.model_dict[model_name].dump_dir = Path(os.path.join(new_lazy_loading_path, 'models', 'ensemble_' + model_name.split('_')[1]))
1497+
model.model_dict[model_name].dump_dir = Path(os.path.join(new_lazy_loading_path, 'models', 'ensemble_' + model_name.split('_')[0]))
14981498

14991499
if remove_original_file:
15001500
os.remove(tar_gz_file)

0 commit comments

Comments
 (0)