works?

mosaicml · Sep 25, 2024 · 4ab5bc1 · 4ab5bc1
1 parent 8cf5473
commit 4ab5bc1
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/llmfoundry/command_utils/data_prep/convert_delta_to_json.py b/llmfoundry/command_utils/data_prep/convert_delta_to_json.py
@@ -547,8 +547,13 @@ def validate_and_get_cluster_info(
         if res is None:
             raise ClusterDoesNotExistError(cluster_id)
 
-        print(res)
-        time.sleep(10)
+        data_security_mode = str(res.data_security_mode).upper()[len('DATASECURITYMODE.'):]
+
+        # USER_ISOLATION stands for Shared Access Mode
+        if data_security_mode not in ('USER_ISOLATION', 'SINGLE_USER'):
+            raise ClusterInvalidAccessMode(
+                cluster_id=cluster_id, access_mode=data_security_mode
+            )
 
         assert res.spark_version is not None
         stripped_runtime = re.sub(