fix: custom writer

CLARIN-PL · Nov 27, 2023 · 90d3128 · 90d3128
1 parent e7cba38
commit 90d3128
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/embeddings/task/lightning_task/lightning_task.py b/embeddings/task/lightning_task/lightning_task.py
@@ -94,7 +94,7 @@ def best_validation_score(self) -> Optional[float]:
 
     def _get_callbacks(self, dataset_subsets: Sequence[str]) -> List[Callback]:
         self.predpath = self.output_path.joinpath("predictions")
-        self.predpath.mkdir(parents=False, exist_ok=False)
+        self.predpath.mkdir(parents=False, exist_ok=True)
         dirpath = self.output_path.joinpath("checkpoints")
         callbacks: List[Callback] = [
             ModelCheckpoint(