podstack 1.3.3__tar.gz → 1.3.5__tar.gz

{podstack-1.3.3 → podstack-1.3.5}/PKG-INFO

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

{podstack-1.3.3 → podstack-1.3.5}/podstack/gpu_runner.py

@@ -812,6 +812,11 @@ _stream_install(
                     raise ConnectionError(f"Failed to get result after {max_retries} attempts: {e}")
                 time.sleep(2 * (attempt + 1))
 
+        # Prefer the streamed output over the API result's output field,
+        # since __PODSTACK_RESULT__ is printed to stdout during execution.
+        if output_buffer:
+            result.output = "".join(output_buffer)
+
         if result.success:
             print(f"\n[Podstack] Completed in {result.gpu_seconds:.1f}s (cost: ₹{result.cost_paise/100:.2f})")
         else:

{podstack-1.3.3 → podstack-1.3.5}/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.3 → podstack-1.3.5}/podstack.egg-info/PKG-INFO

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

{podstack-1.3.3 → podstack-1.3.5}/pyproject.toml

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