bugfixed is_lazy

ersilia-os · Nov 19, 2022 · a194115 · a194115
1 parent e5ed5c6
commit a194115
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/zairachem/__init__.py b/zairachem/__init__.py
@@ -103,7 +103,6 @@ def is_lazy(self):
         model_dir = self.get_trained_dir()
         with open(os.path.join(output_dir, DATA_SUBFOLDER, PRESETS_FILENAME), "r") as f:
             data = json.load(f)
-        print(data)
         if data["is_lazy"]:
             return True
         with open(os.path.join(model_dir, DATA_SUBFOLDER, PRESETS_FILENAME), "r") as f:

diff --git a/zairachem/setup/training.py b/zairachem/setup/training.py
@@ -25,6 +25,7 @@
 from ..vars import APPLICABILITY_SUBFOLDER
 from ..vars import REPORT_SUBFOLDER
 from ..vars import OUTPUT_FILENAME
+from ..vars import PRESETS_FILENAME
 
 from ..tools.melloddy.pipeline import MelloddyTunerTrainPipeline
 from ..augmentation.augment import Augmenter
@@ -78,6 +79,9 @@ def __init__(
         self.time_budget = time_budget  # TODO
 
     def is_lazy(self):
+        with open(os.path.join(self.output_dir, DATA_SUBFOLDER, PRESETS_FILENAME), "w") as f:
+            data = {"is_lazy": self._is_lazy}
+            json.dump(data, f, indent=4)
         return self._is_lazy
 
     def _copy_input_file(self):