tweaks

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.

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.