diff --git a/mlos_bench/mlos_bench/storage/base_storage.py b/mlos_bench/mlos_bench/storage/base_storage.py index 41c9df0e5e..cd529c730c 100644 --- a/mlos_bench/mlos_bench/storage/base_storage.py +++ b/mlos_bench/mlos_bench/storage/base_storage.py @@ -167,7 +167,9 @@ def __exit__( else: assert exc_type and exc_val _LOG.warning( - "Finishing experiment: %s", self, exc_info=(exc_type, exc_val, exc_tb) + "Finishing experiment: %s", + self, + exc_info=(exc_type, exc_val, exc_tb), ) assert self._in_context self._teardown(is_ok) @@ -279,7 +281,10 @@ def load( @abstractmethod def pending_trials( - self, timestamp: datetime, *, running: bool + self, + timestamp: datetime, + *, + running: bool, ) -> Iterator["Storage.Trial"]: """ Return an iterator over the pending trials that are scheduled to run on or @@ -430,7 +435,10 @@ def update( @abstractmethod def update_telemetry( - self, status: Status, timestamp: datetime, metrics: List[Tuple[datetime, str, Any]] + self, + status: Status, + timestamp: datetime, + metrics: List[Tuple[datetime, str, Any]], ) -> None: """ Save the experiment's telemetry data and intermediate status. diff --git a/mlos_bench/mlos_bench/storage/storage_factory.py b/mlos_bench/mlos_bench/storage/storage_factory.py index 2de66a9aab..ea0201717d 100644 --- a/mlos_bench/mlos_bench/storage/storage_factory.py +++ b/mlos_bench/mlos_bench/storage/storage_factory.py @@ -12,7 +12,9 @@ def from_config( - config_file: str, global_configs: Optional[List[str]] = None, **kwargs: Any + config_file: str, + global_configs: Optional[List[str]] = None, + **kwargs: Any, ) -> Storage: """ Create a new storage object from JSON5 config file.