dayhoff-tools 1.13.9__tar.gz → 1.13.11__tar.gz

{dayhoff_tools-1.13.9 → dayhoff_tools-1.13.11}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dayhoff-tools
-Version: 1.13.9
+Version: 1.13.11
 Summary: Common tools for all the repos at Dayhoff Labs
 Author: Daniel Martin-Alarcon
 Author-email: dma@dayhofflabs.com

{dayhoff_tools-1.13.9 → dayhoff_tools-1.13.11}/dayhoff_tools/cli/engines_studios/studio_commands.py

@@ -225,12 +225,14 @@ def studio_status(user: Optional[str], env: Optional[str]):
         # Attached to in blue (if present) - resolve instance ID to engine name
         if studio.get("attached_to"):
             instance_id = studio["attached_to"]
-            # Try to resolve instance ID to engine name
+            # Try to resolve instance ID to engine name by searching engines list
             engine_name = instance_id  # Default to instance ID if not found
             try:
-                engine = client.get_engine_by_name(instance_id)
-                if engine:
-                    engine_name = engine.get("name", instance_id)
+                engines_result = client.list_engines()
+                for engine in engines_result.get("engines", []):
+                    if engine.get("instance_id") == instance_id:
+                        engine_name = engine.get("name", instance_id)
+                        break
             except Exception:
                 pass  # Fall back to instance ID
             click.echo(f"Attached to: \033[34m{engine_name}\033[0m")

{dayhoff_tools-1.13.9 → dayhoff_tools-1.13.11}/dayhoff_tools/deployment/deploy_aws.py

@@ -344,46 +344,55 @@ def create_or_update_job_definition(
             f"Adding mount points to job definition: {batch_job_config['mountPoints']}"
         )
 
-    # Auto-mount Primordial Drive if available
-    # This ensures all batch jobs have access to shared datasets at /primordial/
-    primordial_fs_id = get_primordial_fs_id(session)
-    if primordial_fs_id:
-        print(f"Adding Primordial Drive configuration (fs_id: {primordial_fs_id})")
-
-        # Add volume configuration
-        efs_volume = {
-            "name": "primordial",
-            "efsVolumeConfiguration": {
-                "fileSystemId": primordial_fs_id,
-                "rootDirectory": "/",
-            },
-        }
-
-        if "volumes" not in container_properties:
-            container_properties["volumes"] = []
-
-        # Check if already added to avoid duplicates
-        if not any(
-            v.get("name") == "primordial" for v in container_properties["volumes"]
-        ):
-            container_properties["volumes"].append(efs_volume)
-
-        # Add mount point
-        mount_point = {
-            "sourceVolume": "primordial",
-            "containerPath": "/primordial",
-            "readOnly": False,
-        }
-
-        if "mountPoints" not in container_properties:
-            container_properties["mountPoints"] = []
-
-        # Check if already added
-        if not any(
-            mp.get("containerPath") == "/primordial"
-            for mp in container_properties["mountPoints"]
-        ):
-            container_properties["mountPoints"].append(mount_point)
+    # Mount Primordial Drive if explicitly enabled via feature flag
+    # Add 'mount_primordial_drive' to features list to mount shared EFS at /primordial/
+    features = config.get("features", []) or []
+    features_set = {f if isinstance(f, str) else next(iter(f)) for f in features}
+    mount_primordial = "mount_primordial_drive" in features_set
+
+    if mount_primordial:
+        primordial_fs_id = get_primordial_fs_id(session)
+        if primordial_fs_id:
+            print(f"Adding Primordial Drive configuration (fs_id: {primordial_fs_id})")
+
+            # Add volume configuration
+            efs_volume = {
+                "name": "primordial",
+                "efsVolumeConfiguration": {
+                    "fileSystemId": primordial_fs_id,
+                    "rootDirectory": "/",
+                },
+            }
+
+            if "volumes" not in container_properties:
+                container_properties["volumes"] = []
+
+            # Check if already added to avoid duplicates
+            if not any(
+                v.get("name") == "primordial" for v in container_properties["volumes"]
+            ):
+                container_properties["volumes"].append(efs_volume)
+
+            # Add mount point
+            mount_point = {
+                "sourceVolume": "primordial",
+                "containerPath": "/primordial",
+                "readOnly": False,
+            }
+
+            if "mountPoints" not in container_properties:
+                container_properties["mountPoints"] = []
+
+            # Check if already added
+            if not any(
+                mp.get("containerPath") == "/primordial"
+                for mp in container_properties["mountPoints"]
+            ):
+                container_properties["mountPoints"].append(mount_point)
+        else:
+            print(
+                "Warning: mount_primordial_drive enabled but Primordial Drive not found in this environment"
+            )
 
     # Check if job definition already exists using the session client
     try:

{dayhoff_tools-1.13.9 → dayhoff_tools-1.13.11}/pyproject.toml

@@ -5,7 +5,7 @@ build-backend = "poetry.core.masonry.api"
 
 [project]
 name = "dayhoff-tools"
-version = "1.13.9"
+version = "1.13.11"
 description = "Common tools for all the repos at Dayhoff Labs"
 authors = [
     {name = "Daniel Martin-Alarcon", email = "dma@dayhofflabs.com"}