fix(automl/tuners): ensure that the checkpoint file parents exist

ruancomelli · Jul 3, 2022 · 2d92445 · 2d92445
1 parent 5b9fe5f
commit 2d92445
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/boiling_learning/automl/tuners.py b/boiling_learning/automl/tuners.py
@@ -81,7 +81,9 @@ def _build_and_fit_model(
         )
 
     def _get_checkpoint_fname(self, trial_id: str) -> str:
-        return str(resolve(super()._get_checkpoint_fname(trial_id)).with_suffix('.h5'))
+        return str(
+            resolve(super()._get_checkpoint_fname(trial_id), parents=True).with_suffix('.h5')
+        )
 
 
 class _FixedMaxModelSizeGreedy(_SaveBestModelAtTrainingEnd):