podstack 1.3.17__tar.gz → 1.3.18__tar.gz

{podstack-1.3.17 → podstack-1.3.18}/PKG-INFO

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

{podstack-1.3.17 → podstack-1.3.18}/podstack/annotations.py

@@ -90,6 +90,7 @@ class GPUConfig:
         conda: Union[str, list] = None,
         requirements: str = None,
         use_uv: bool = False,
+        env_vars: Optional[Dict[str, str]] = None,
         remote: bool = None
     ):
         """
@@ -121,6 +122,7 @@ class GPUConfig:
         self.conda = conda
         self.requirements = requirements
         self.use_uv = use_uv
+        self.env_vars = env_vars or {}
         self.remote = remote if remote is not None else _remote_execution_enabled
 
         # Store in global config
@@ -138,6 +140,7 @@ class GPUConfig:
             "conda": conda,
             "requirements": requirements,
             "use_uv": use_uv,
+            "env_vars": env_vars or {},
         }
 
     def __call__(self, func: Callable) -> Callable:
@@ -249,6 +252,7 @@ if __podstack_result__ is not None:
                     conda=effective_conda,
                     requirements=self.requirements,
                     use_uv=self.use_uv,
+                    env_vars=self.env_vars,
                     runner=runner_name,
                     wait=True,
                     stream=None  # Auto-detect: True in Jupyter, False otherwise
@@ -340,6 +344,7 @@ def gpu(
     conda: Union[str, list] = None,
     requirements: str = None,
     use_uv: bool = False,
+    env_vars: Optional[Dict[str, str]] = None,
     remote: bool = None
 ) -> GPUConfig:
     """
@@ -421,6 +426,7 @@ def gpu(
         conda=conda,
         requirements=requirements,
         use_uv=use_uv,
+        env_vars=env_vars,
         remote=remote
     )
 

{podstack-1.3.17 → podstack-1.3.18}/podstack/gpu_runner.py

@@ -361,6 +361,7 @@ class GPURunner:
         conda: Union[str, list] = None,
         requirements: str = None,
         use_uv: bool = False,
+        env_vars: Dict[str, str] = None,
         add_annotation: bool = True,
         runner: str = None
     ) -> Dict[str, Any]:
@@ -395,6 +396,13 @@ class GPURunner:
             annotation = self._build_annotation(gpu, count, fraction, timeout, env, pip, uv, conda, requirements, use_uv, runner)
             code = f"{annotation}\n\n{code}"
 
+        # Inject environment variables
+        if env_vars:
+            env_lines = ["import os"]
+            for k, v in env_vars.items():
+                env_lines.append(f"os.environ[{repr(k)}] = {repr(str(v))}")
+            code = "\n".join(env_lines) + "\n\n" + code
+
         # Build installation code for packages
         install_parts = []
 
@@ -672,6 +680,7 @@ _stream_install(
         conda: Union[str, list] = None,
         requirements: str = None,
         use_uv: bool = False,
+        env_vars: Dict[str, str] = None,
         wait: bool = True,
         poll_interval: float = 2.0,
         max_retries: int = 3,
@@ -713,7 +722,7 @@ _stream_install(
             ValueError: If parameters are invalid
         """
         # Submit the code
-        submission = self.submit(code, gpu, count, fraction, timeout, env, pip, uv, conda, requirements, use_uv, runner=runner)
+        submission = self.submit(code, gpu, count, fraction, timeout, env, pip, uv, conda, requirements, use_uv, env_vars=env_vars, runner=runner)
         execution_id = submission.get("execution_id")
 
         if not execution_id:

{podstack-1.3.17 → podstack-1.3.18}/podstack/registry/__init__.py

@@ -558,9 +558,23 @@ def create_trial(sweep_id: str, run_id: str, params: dict) -> dict:
     return _get_client().create_trial(sweep_id, run_id, params)
 
 
-def complete_trial(sweep_id: str, trial_id: str, value: float) -> None:
-    """Mark a trial as completed with its objective metric value."""
-    _get_client().complete_trial(sweep_id, trial_id, value)
+def complete_trial(
+    sweep_id: str,
+    trial_id: str,
+    value: float = None,
+    metrics: dict = None,
+    run_id: str = None,
+) -> None:
+    """Mark a trial as completed with its objective metric value.
+
+    Args:
+        sweep_id: Sweep ID.
+        trial_id: Trial ID (from suggest_trial_params).
+        value: Scalar objective value.
+        metrics: Dict of metric name → value; first entry used if value not set.
+        run_id: Optional run ID to link to this trial.
+    """
+    _get_client().complete_trial(sweep_id, trial_id, value, metrics, run_id)
 
 
 def list_trials(sweep_id: str) -> list:

{podstack-1.3.17 → podstack-1.3.18}/podstack/registry/client.py

@@ -1372,9 +1372,12 @@ class RegistryClient:
         return data.get("sweeps", [])
 
     def suggest_trial_params(self, sweep_id: str) -> Dict[str, Any]:
-        """Get suggested hyperparameter values for the next trial."""
+        """Get suggested hyperparameter values for the next trial.
+
+        Returns a dict with ``id`` (trial ID) and ``params`` (hyperparameter map).
+        """
         data = self._request("GET", f"/sweeps/{sweep_id}/suggest")
-        return data.get("params", data)
+        return data.get("trial", data)
 
     def create_trial(
         self,
@@ -1389,11 +1392,31 @@ class RegistryClient:
         })
         return data.get("trial", data)
 
-    def complete_trial(self, sweep_id: str, trial_id: str, value: float) -> None:
-        """Mark a trial as completed with its objective metric value."""
-        self._request("POST", f"/sweeps/{sweep_id}/trials/{trial_id}/complete", json={
-            "value": value
-        })
+    def complete_trial(
+        self,
+        sweep_id: str,
+        trial_id: str,
+        value: float = None,
+        metrics: Dict[str, float] = None,
+        run_id: str = None,
+    ) -> None:
+        """Mark a trial as completed with its objective metric value.
+
+        Args:
+            sweep_id: Sweep ID.
+            trial_id: Trial ID (from suggest_trial_params).
+            value: Scalar objective value.
+            metrics: Dict of metric name → value; first entry used if value not set.
+            run_id: Optional run ID to link to this trial.
+        """
+        body: Dict[str, Any] = {}
+        if value is not None:
+            body["value"] = value
+        if metrics:
+            body["metrics"] = metrics
+        if run_id:
+            body["run_id"] = run_id
+        self._request("POST", f"/sweeps/{sweep_id}/trials/{trial_id}/complete", json=body)
 
     def list_trials(self, sweep_id: str) -> List[Dict[str, Any]]:
         """List all trials for a sweep."""

{podstack-1.3.17 → podstack-1.3.18}/podstack.egg-info/PKG-INFO

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

{podstack-1.3.17 → podstack-1.3.18}/pyproject.toml

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