Use the is_read_only property to determine read-only mode

arkindex_worker/worker/training.py

@@ -102,12 +102,12 @@ def build_clean_payload(**kwargs):
 
 def skip_if_read_only(func):
     """
-    Return shortly in case the read_only attribute is evaluated to True
+    Return shortly in case the is_read_only property is evaluated to True
     """
 
     @functools.wraps(func)
     def wrapper(self, *args, **kwargs):
-        if getattr(self, "read_only", False):
+        if getattr(self, "is_read_only", False):
             logger.warning(
                 "Cannot perform this operation as the worker is in read-only mode"
             )

tests/test_elements_worker/test_training.py

@@ -101,7 +101,10 @@ def test_handle_s3_uploading_errors(mock_training_worker, model_file_dir):
 )
 def test_training_mixin_read_only(mock_training_worker, method, caplog):
     """All operations related to models versions returns early if the worker is configured as read only"""
-    mock_training_worker.read_only = True
+    # Set worker in read_only mode
+    mock_training_worker.worker_run_id = None
+    assert mock_training_worker.is_read_only
+
     assert mock_training_worker.model_version is None
     getattr(mock_training_worker, method)()
     assert mock_training_worker.model_version is None