diff --git a/simvue/run.py b/simvue/run.py index 77e6dcd9..4acf44e4 100644 --- a/simvue/run.py +++ b/simvue/run.py @@ -921,7 +921,7 @@ def _add_metrics_to_dispatch( self, metrics: dict[str, typing.Union[int, float]], step: typing.Optional[int] = None, - time: typing.Optional[int] = None, + time: typing.Optional[float] = None, timestamp: typing.Optional[str] = None, join_on_fail: bool = True, ) -> bool: @@ -967,7 +967,7 @@ def log_metrics( self, metrics: dict[str, typing.Union[int, float]], step: typing.Optional[int] = None, - time: typing.Optional[int] = None, + time: typing.Optional[float] = None, timestamp: typing.Optional[str] = None, ) -> bool: """