Commit 9d7ef0e7 authored by Charly Lamothe's avatar Charly Lamothe
Browse files

Fix on resume mode for extracted forest size job

parent c43ae6c4
...@@ -141,7 +141,7 @@ def extracted_forest_size_job(extracted_forest_size_job_pb, extracted_forest_siz ...@@ -141,7 +141,7 @@ def extracted_forest_size_job(extracted_forest_size_job_pb, extracted_forest_siz
sub_models_dir_files = os.listdir(sub_models_dir) sub_models_dir_files = os.listdir(sub_models_dir)
for file_name in sub_models_dir_files: for file_name in sub_models_dir_files:
if '.pickle' != os.path.splitext(file_name)[1]: if '.pickle' != os.path.splitext(file_name)[1]:
return continue
else: else:
already_exists = os.path.getsize(os.path.join(sub_models_dir, file_name)) > 0 already_exists = os.path.getsize(os.path.join(sub_models_dir, file_name)) > 0
break break
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment