simulo 0.31.0__tar.gz → 0.31.1__tar.gz

{simulo-0.31.0/src/simulo.egg-info → simulo-0.31.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: simulo
-Version: 0.31.0
+Version: 0.31.1
 Summary: Python SDK and CLI for cloud-native robotics simulation, reinforcement learning, and robot policy training on managed GPUs.
 Author-email: Simulo Team <team@simulo.ai>
 License: BSD-3-Clause

{simulo-0.31.0 → simulo-0.31.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "simulo"
-version = "0.31.0"
+version = "0.31.1"
 description = "Python SDK and CLI for cloud-native robotics simulation, reinforcement learning, and robot policy training on managed GPUs."
 readme = "PYPI.md"
 requires-python = ">=3.11"

{simulo-0.31.0 → simulo-0.31.1}/src/simulo/_client/cli.py

@@ -673,6 +673,14 @@ def _resolve_client_context() -> tuple[str, Optional[str]]:
     return base_url, token
 
 
+def _client_with_fresh_token(client: JobsApiClient) -> JobsApiClient:
+    """Return a client with current stored credentials, preserving its target."""
+    creds = credentials.try_get_valid_credentials(base_url=client.base_url)
+    if creds is None:
+        return client
+    return JobsApiClient(client.base_url, token=str(creds["access_token"]))
+
+
 def _is_uuid_text(value: Any) -> bool:
     try:
         UUID(str(value))
@@ -1470,10 +1478,16 @@ def _follow_and_report(handle: JobHandle, app_path: str) -> int:
         job_ref = str(getattr(handle, "public_id", None) or handle.job_id)
         exit_code = _drain_logs(client, job_ref, follow=True, scope=JOB_SCOPE_MINE)
         if handle.job_type is JobType.PREVIEW:
+            client = _client_with_fresh_token(client)
             return _report_preview(client, job_ref, display_id, app_path, fallback_exit_code=exit_code)
         if exit_code == 0:
             if handle.job_type is JobType.TRAIN:
-                _print_training_next_steps(handle, app_path)
+                client = _client_with_fresh_token(client)
+                _print_training_next_steps(
+                    PoliciesApiClient(client.base_url, token=client._token),
+                    handle,
+                    app_path,
+                )
             elif handle.job_type is JobType.PLAY:
                 print(f"Next: simulo recordings {display_id}")
         return exit_code
@@ -1566,12 +1580,11 @@ def _checkpoint_iteration(handle: JobHandle, checkpoint_ref: str) -> Optional[in
     return None
 
 
-def _print_training_next_steps(handle: JobHandle, app_path: str) -> None:
+def _print_training_next_steps(client: PoliciesApiClient, handle: JobHandle, app_path: str) -> None:
     """Print the sealed policy's checkpoints and the three canonical next steps."""
-    assert handle.base_url is not None
     policy_id = str(policy_id_for_job(_handle_display_id(handle)))
     try:
-        record = PoliciesApiClient(handle.base_url, token=handle.token).get_policy(policy_id)
+        record = client.get_policy(policy_id)
     except http.HttpError:
         return
     rendered: list[str] = []
@@ -2386,6 +2399,7 @@ def _wait_for_job(client: JobsApiClient, job_id: str, *, scope: str = JOB_SCOPE_
     """Poll until ``job_id`` exists — lets a user arm the tail BEFORE the job is created."""
     waiting_announced = False
     while True:
+        client = _client_with_fresh_token(client)
         try:
             return client.get_job(job_id, scope=scope)
         except JobsApiHTTPError as exc:
@@ -2414,6 +2428,7 @@ def _wait_for_latest_job(client: JobsApiClient) -> dict[str, Any]:
     """
     waiting_announced = False
     while True:
+        client = _client_with_fresh_token(client)
         try:
             latest = client.get_latest_job()
         except JobsApiHTTPError as exc:
@@ -2608,6 +2623,8 @@ def _drain_logs(
     post-``simulo run`` default follow — one follow loop, one
     exit-code-mirrors-outcome rule.
     """
+    if follow:
+        client = _client_with_fresh_token(client)
     offset = 0
     if tail_lines is not None:
         try:
@@ -2633,6 +2650,8 @@ def _drain_logs(
         # Not honored (old server): the loop below simply continues the full
         # history from where the server's offset-0 answer left off.
     while True:
+        if follow:
+            client = _client_with_fresh_token(client)
         try:
             chunk, offset, status, _ = client.get_log_chunk(job_id, offset, scope=scope)
         except JobsApiHTTPError as exc:
@@ -2770,9 +2789,9 @@ def _cmd_result(args: argparse.Namespace) -> int:
     announced on stderr.
     A job whose run announced typed outputs gets a second
     stderr line, ``⚠ 1 anomaly (non_finite_reward)`` (:func:`_outputs_flag_summary`)
-    — the result JSON itself is untouched (``stats["anomaly_captures"]`` stays
-    byte-identical; a byte stream is an output, a JSON scalar is a metric,
-    surfaced together but never merged). Both lines print to
+    while built-in training results omit anomaly-capture paths. Those paths
+    are exposed through the recordings and outputs commands instead. Both
+    lines print to
     stderr so stdout stays pure result JSON for pipelines. Best-effort: for
     an explicit JOB_ID this needs one extra ``get_job`` call, and a failure
     there (e.g. an older control plane) skips both lines rather than failing a

{simulo-0.31.0 → simulo-0.31.1/src/simulo.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: simulo
-Version: 0.31.0
+Version: 0.31.1
 Summary: Python SDK and CLI for cloud-native robotics simulation, reinforcement learning, and robot policy training on managed GPUs.
 Author-email: Simulo Team <team@simulo.ai>
 License: BSD-3-Clause