bugfix (#108)

premAI-io · Apr 19, 2024 · 1432afd · 1432afd
1 parent d563af2
commit 1432afd
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/prem_utils/connectors/anyscale.py b/prem_utils/connectors/anyscale.py
@@ -83,6 +83,9 @@ def create_job(
         validation_dataset: list[Datapoint] | None = None,
         num_epochs: int = 3,
     ) -> str:
+        if "anyscale" in model:
+            model = model.replace("anyscale/", "", 1)
+
         training_file_id = self._upload_data(training_dataset, size=20)
 
         validation_file_id = None