Disable training methods usage in read only mode

Closes #142 (closed)

arkindex_worker/worker/training.py

@@ -87,6 +87,11 @@ class TrainingMixin(object):
         This method creates a model archive and its associated hash,
         to create a unique version that will be stored on a bucket and published on arkindex.
         """
+        if self.is_read_only:
+            logger.warning(
+                "Cannot publish a new model version as this worker is in read-only mode"
+            )
+            return
 
         # Create the zst archive, get its hash and size
         with create_archive(path=model_path) as (
@@ -131,6 +136,11 @@ class TrainingMixin(object):
         - The version is in `Created` state: this version's details is used
         - The version is in `Available` state: you cannot create twice the same version, an error is raised
         """
+        if self.is_read_only:
+            logger.warning(
+                "Cannot create a new model version as this worker is in read-only mode"
+            )
+            return
 
         # Create a new model version with hash and size
         try:
@@ -171,6 +181,11 @@ class TrainingMixin(object):
         """
         Upload the archive of the model's files to an Amazon s3 compatible storage
         """
+        if self.is_read_only:
+            logger.warning(
+                "Cannot upload this archive as this worker is in read-only mode"
+            )
+            return
 
         s3_put_url = model_version_details.get("s3_put_url")
         logger.info("Uploading to s3...")
@@ -191,6 +206,11 @@ class TrainingMixin(object):
         """
         Update the specified model version to the state `Available` and use the given information"
         """
+        if self.is_read_only:
+            logger.warning(
+                "Cannot update this model version as this worker is in read-only mode"
+            )
+            return
 
         model_version_id = model_version_details.get("id")
         logger.info(f"Updating model version ({model_version_id})")