podstack 1.3.2__tar.gz → 1.3.4__tar.gz

{podstack-1.3.2 → podstack-1.3.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: podstack
-Version: 1.3.2
+Version: 1.3.4
 Summary: Official Python SDK for Podstack GPU Notebook Platform
 Author-email: Podstack <support@podstack.ai>
 License-Expression: MIT

{podstack-1.3.2 → podstack-1.3.4}/podstack/annotations.py

@@ -142,6 +142,12 @@ class GPUConfig:
 
     def __call__(self, func: Callable) -> Callable:
         """Decorator usage - executes function on remote GPU."""
+        # Capture runner config at decoration time so it survives across
+        # multiple calls (the global is cleared after the first call).
+        global _current_runner_config
+        _captured_runner_config = _current_runner_config
+        _current_runner_config = None  # clear here, not inside wrapper
+
         @functools.wraps(func)
         def wrapper(*args, **kwargs):
             if not self.remote:
@@ -213,22 +219,20 @@ if __podstack_result__ is not None:
     print(repr(__podstack_result__))
 """
 
-            # Check for active runner config from @podstack.runner() decorator
-            global _current_runner_config
+            # Use the runner config captured at decoration time
             runner_name = None
             effective_pip = self.pip
             effective_uv = self.uv
             effective_conda = self.conda
-            if _current_runner_config is not None:
-                runner_name = _current_runner_config.get("image")
+            if _captured_runner_config is not None:
+                runner_name = _captured_runner_config.get("image")
                 # Merge package lists from runner config
-                if _current_runner_config.get("pip") and not effective_pip:
-                    effective_pip = _current_runner_config["pip"]
-                if _current_runner_config.get("uv") and not effective_uv:
-                    effective_uv = _current_runner_config["uv"]
-                if _current_runner_config.get("conda") and not effective_conda:
-                    effective_conda = _current_runner_config["conda"]
-                _current_runner_config = None  # Clear after use
+                if _captured_runner_config.get("pip") and not effective_pip:
+                    effective_pip = _captured_runner_config["pip"]
+                if _captured_runner_config.get("uv") and not effective_uv:
+                    effective_uv = _captured_runner_config["uv"]
+                if _captured_runner_config.get("conda") and not effective_conda:
+                    effective_conda = _captured_runner_config["conda"]
 
             # Execute on remote GPU with improved error handling
             # Auto-enable streaming in Jupyter notebooks for real-time output

{podstack-1.3.2 → podstack-1.3.4}/podstack/gpu_runner.py

@@ -211,7 +211,7 @@ class GPUExecutionResult:
             output=data.get("output", ""),
             error=data.get("error"),
             gpu_seconds=data.get("gpu_seconds", 0.0),
-            cost_paise=data.get("actual_cost_paise", 0),
+            cost_paise=data.get("cost_cents", data.get("cost_paise", data.get("actual_cost_paise", 0))),
             gpu_type=data.get("gpu_type", ""),
             gpu_count=data.get("gpu_count", 1),
             success=data.get("status") == "completed"

{podstack-1.3.2 → podstack-1.3.4}/podstack/registry/experiment.py

@@ -197,6 +197,11 @@ class Run:
         if self._client:
             self._client.log_artifact(local_path, artifact_path)
 
+    def log_model(self, model, artifact_path: str = "model", framework: str = None, metadata: Dict[str, str] = None):
+        """Serialize and log a model for this run."""
+        if self._client:
+            self._client.log_model(model, artifact_path=artifact_path, framework=framework, metadata=metadata)
+
     def set_tag(self, key: str, value: str):
         """Set a tag on this run."""
         if self._client:
@@ -209,6 +214,10 @@ class Run:
             self._client.end_run(status)
         self.status = status
 
+    def end_run(self, status: str = "completed"):
+        """End this run (alias for end())."""
+        self.end(status)
+
 
 def _parse_datetime(value: Any) -> Optional[datetime]:
     """Parse a datetime from string or return None."""

{podstack-1.3.2 → podstack-1.3.4}/podstack.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: podstack
-Version: 1.3.2
+Version: 1.3.4
 Summary: Official Python SDK for Podstack GPU Notebook Platform
 Author-email: Podstack <support@podstack.ai>
 License-Expression: MIT

{podstack-1.3.2 → podstack-1.3.4}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "podstack"
-version = "1.3.2"
+version = "1.3.4"
 description = "Official Python SDK for Podstack GPU Notebook Platform"
 readme = "README.md"
 license = "MIT"