computefence 0.2.8__tar.gz → 0.2.9__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: computefence
-Version: 0.2.8
+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.8 → 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.8 → computefence-0.2.9}/computefence.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: computefence
-Version: 0.2.8
+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.8/computefence/__init__.py

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