From 9d7ef0e707a256c8ba04b13933b3af017a10bcd6 Mon Sep 17 00:00:00 2001
From: Charly Lamothe <charly.lamothe@univ-amu.fr>
Date: Thu, 12 Mar 2020 17:37:44 +0100
Subject: [PATCH] Fix on resume mode for extracted forest size job
---
code/train.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/code/train.py b/code/train.py
index 8e48e14..ff0e9e0 100644
--- a/code/train.py
+++ b/code/train.py
@@ -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)
for file_name in sub_models_dir_files:
if '.pickle' != os.path.splitext(file_name)[1]:
- return
+ continue
else:
already_exists = os.path.getsize(os.path.join(sub_models_dir, file_name)) > 0
break
--
GitLab