dayhoff-tools 1.2.3__tar.gz → 1.2.4__tar.gz

{dayhoff_tools-1.2.3 → dayhoff_tools-1.2.4}/PKG-INFO

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

{dayhoff_tools-1.2.3 → dayhoff_tools-1.2.4}/dayhoff_tools/warehouse.py

@@ -282,6 +282,27 @@ def add_to_warehouse(
     # Change the working directory to the warehouse folder
     os.chdir("warehouse")
 
+    # Configure DVC cache for Lightning Studio if needed
+    if os.path.exists("/teamspace/studios/this_studio"):
+        studio_cache_dir = "/teamspace/studios/this_studio/.dvc_cache"
+        os.makedirs(studio_cache_dir, exist_ok=True)
+        try:
+            result = subprocess.run(
+                ["dvc", "cache", "dir"], capture_output=True, text=True, check=True
+            )
+            current_cache = result.stdout.strip()
+            if current_cache != studio_cache_dir:
+                subprocess.run(
+                    ["dvc", "cache", "dir", studio_cache_dir, "--local"], check=True
+                )
+                print(
+                    f"✅ Configured warehouse DVC cache for Lightning Studio: {studio_cache_dir}"
+                )
+        except subprocess.CalledProcessError:
+            subprocess.run(
+                ["dvc", "cache", "dir", studio_cache_dir, "--local"], check=True
+            )
+
     # Add and push the data file
     subprocess.run(["dvc", "add", warehouse_path], check=True)
 

{dayhoff_tools-1.2.3 → dayhoff_tools-1.2.4}/pyproject.toml

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