computefence 0.2.7__tar.gz → 0.2.9__tar.gz

{computefence-0.2.7 → computefence-0.2.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: computefence
-Version: 0.2.7
+Version: 0.2.9
 Summary: Don't lose another RunPod checkpoint to ephemeral disk before your job bills.
 License-Expression: MIT
 Requires-Python: >=3.8

computefence-0.2.9/computefence/__init__.py

@@ -0,0 +1 @@
+__version__ = "0.2.9"

{computefence-0.2.7 → computefence-0.2.9}/computefence/checks/storage.py

@@ -33,7 +33,7 @@ def check_storage():
                 "message": f"HF_HOME is not set. Mounted volume detected at {mounted_volumes[0]} but HuggingFace will cache to default local path.",
                 "fix": f"export HF_HOME={mounted_volumes[0]}/.cache/huggingface"
             })
-       else:
+        else:
             results.append({
                 "status": "warn",
                 "message": "HF_HOME is not set. HuggingFace will use default local cache.",

{computefence-0.2.7 → computefence-0.2.9}/computefence/telemetry.py

@@ -48,7 +48,15 @@ def record_run(results: list):
         import torch
         gpu_count = torch.cuda.device_count()
     except Exception:
-        gpu_count = 0
+        try:
+            import subprocess
+            result = subprocess.run(
+                ["nvidia-smi", "--query-gpu=name", "--format=csv,noheader"],
+                capture_output=True, text=True, timeout=5
+            )
+            gpu_count = len(result.stdout.strip().splitlines()) if result.returncode == 0 else 0
+        except Exception:
+            gpu_count = 0
 
     from computefence import __version__
 

{computefence-0.2.7 → computefence-0.2.9}/computefence.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: computefence
-Version: 0.2.7
+Version: 0.2.9
 Summary: Don't lose another RunPod checkpoint to ephemeral disk before your job bills.
 License-Expression: MIT
 Requires-Python: >=3.8

computefence-0.2.7/computefence/__init__.py

@@ -1 +0,0 @@
-__version__ = "0.2.7"