🐛 Fixed grid metric and span retrieval

Author: kzscisoft

simvue/api/objects/grids.py

@@ -221,28 +221,36 @@ def get_run_metric_span(self, *, run_id: str, metric_name: str) -> dict:
             metric to retrieve span information for.
 
         Returns
-        ------
+        -------
         dict[str, list[dict[str, float]]
             dictionary containing span from this for the run at specified step.
         """
         _response = sv_get(
-            url=f"{self.run_metric_values_url(run_id, metric_name) / 'span'}",
+            url=f"{self.run_metric_url(run_id, metric_name) / 'span'}",
             headers=self._headers,
         )
 
-        return get_json_from_response(
-            response=_response,
-            expected_status=[http.HTTPStatus.OK],
-            expected_type=dict,
-            scenario=f"Retrieving grid span for run '{run_id}'",
+        _response = typing.cast(
+            "list[dict[str, int | float | str]]",
+            get_json_from_response(
+                response=_response,
+                expected_status=[http.HTTPStatus.OK],
+                expected_type=list,
+                scenario=f"Retrieving grid span for run '{run_id}'",
+            ),
         )
 
+        if not _response:
+            raise RuntimeError("Failed to retrieve span, no data.")
+
+        return _response[0]
+
     @classmethod
     def get(
         cls,
         *_,
         **__,
-    ) -> typing.Generator[tuple[str, Self | None], None, None]:
+    ) -> typing.Generator[tuple[str, Self | None]]:
         raise NotImplementedError
 
 
@@ -309,7 +317,7 @@ def get(
         step: pydantic.NonNegativeInt,
         spans: bool = False,
         **kwargs,
-    ) -> typing.Generator[dict[str, dict[str, list[dict[str, float]]]], None, None]:
+    ) -> typing.Generator[dict[str, dict[str, list[dict[str, float]]]]]:
         """Retrieve tensor-metrics from the server for a given set of runs.
 
         Parameters

tests/unit/test_grids.py

@@ -132,6 +132,8 @@ def test_grid_metrics_creation_online() -> None:
     time.sleep(1)
     # Online metrics
     assert list(GridMetrics.get(runs=[_run.id], metrics=["A"], step=_step))
+    assert _grid.get_run_metric_values(run_id=_run.id, metric_name="A", step=_step)
+    assert _grid.get_run_metric_span(run_id=_run.id, metric_name="A").get("max_timestamp")
     _run.delete()
     _folder.delete(recursive=True, delete_runs=True, runs_only=False)