[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

mlos_bench/mlos_bench/services/types/authenticator_type.py

@@ -13,6 +13,7 @@
 @runtime_checkable
 class SupportsAuth(Protocol[T_co]):
     """Protocol interface for authentication for the cloud services."""
+
     # pylint: disable=unnecessary-ellipsis
 
     def get_access_token(self) -> str:

mlos_bench/mlos_bench/services/types/config_loader_type.py

@@ -22,6 +22,7 @@
 @runtime_checkable
 class SupportsConfigLoading(Protocol):
     """Protocol interface for helper functions to lookup and load configs."""
+
     # pylint: disable=unnecessary-ellipsis
 
     def get_config_paths(self) -> list[str]:

mlos_bench/mlos_bench/services/types/host_ops_type.py

@@ -13,6 +13,7 @@
 @runtime_checkable
 class SupportsHostOps(Protocol):
     """Protocol interface for Host/VM boot operations."""
+
     # pylint: disable=unnecessary-ellipsis
 
     def start_host(self, params: dict) -> tuple["Status", dict]:

mlos_bench/mlos_bench/services/types/host_provisioner_type.py

@@ -13,6 +13,7 @@
 @runtime_checkable
 class SupportsHostProvisioning(Protocol):
     """Protocol interface for Host/VM provisioning operations."""
+
     # pylint: disable=unnecessary-ellipsis
 
     def provision_host(self, params: dict) -> tuple["Status", dict]:

mlos_bench/mlos_bench/services/types/local_exec_type.py

@@ -23,6 +23,7 @@ class SupportsLocalExec(Protocol):
     Can be useful for data processing due to reduced dependency management complications
     vs the target environment. Used in LocalEnv and provided by LocalExecService.
     """
+
     # pylint: disable=unnecessary-ellipsis
 
     def local_exec(

mlos_bench/mlos_bench/services/types/network_provisioner_type.py

@@ -13,6 +13,7 @@
 @runtime_checkable
 class SupportsNetworkProvisioning(Protocol):
     """Protocol interface for Network provisioning operations."""
+
     # pylint: disable=unnecessary-ellipsis
 
     def provision_network(self, params: dict) -> tuple["Status", dict]:

mlos_bench/mlos_bench/services/types/os_ops_type.py

@@ -13,6 +13,7 @@
 @runtime_checkable
 class SupportsOSOps(Protocol):
     """Protocol interface for Host/OS operations."""
+
     # pylint: disable=unnecessary-ellipsis
 
     def shutdown(self, params: dict, force: bool = False) -> tuple["Status", dict]:

mlos_bench/mlos_bench/services/types/remote_config_type.py

@@ -13,6 +13,7 @@
 @runtime_checkable
 class SupportsRemoteConfig(Protocol):
     """Protocol interface for configuring cloud services."""
+
     # pylint: disable=unnecessary-ellipsis
 
     def configure(self, config: dict[str, Any], params: dict[str, Any]) -> tuple["Status", dict]:

mlos_bench/mlos_bench/services/types/remote_exec_type.py

@@ -18,6 +18,7 @@ class SupportsRemoteExec(Protocol):
     """Protocol interface for Service types that provide helper functions to run scripts
     on a remote host OS.
     """
+
     # pylint: disable=unnecessary-ellipsis
 
     def remote_exec(

mlos_bench/mlos_bench/services/types/vm_provisioner_type.py

@@ -13,6 +13,7 @@
 @runtime_checkable
 class SupportsVMOps(Protocol):
     """Protocol interface for VM provisioning operations."""
+
     # pylint: disable=unnecessary-ellipsis
 
     def vm_provision(self, params: dict) -> tuple["Status", dict]:

mlos_bench/mlos_bench/tests/environments/composite_env_service_test.py

@@ -5,12 +5,12 @@
 """Check how the services get inherited and overridden in child environments."""
 import os
 
-from mlos_bench.services.local.temp_dir_context import TempDirContextService
 import pytest
 
 from mlos_bench.environments.composite_env import CompositeEnv
 from mlos_bench.services.config_persistence import ConfigPersistenceService
 from mlos_bench.services.local.local_exec import LocalExecService
+from mlos_bench.services.local.temp_dir_context import TempDirContextService
 from mlos_bench.tunables.tunable_groups import TunableGroups
 from mlos_bench.util import path_join
 

mlos_bench/mlos_bench/tests/services/mock_service.py

@@ -17,6 +17,7 @@
 @runtime_checkable
 class SupportsSomeMethod(Protocol):
     """Protocol for some_method."""
+
     # pylint: disable=unnecessary-ellipsis
 
     def some_method(self) -> str:

mlos_bench/mlos_bench/tunables/tunable.py

@@ -6,7 +6,7 @@
 import copy
 import logging
 from collections.abc import Iterable, Sequence
-from typing import Any, Literal, TypedDict, Required
+from typing import Any, Literal, Required, TypedDict
 
 import numpy as np
 
@@ -69,7 +69,8 @@ class TunableDict(TypedDict, total=False):
 
 
 def tunable_dict_from_dict(config: dict[str, Any]) -> "TunableDict":
-    """Creates a TunableDict from a regular dict.
+    """
+    Creates a TunableDict from a regular dict.
 
     Parameters
     ----------

mlos_core/mlos_core/spaces/converters/flaml.py

@@ -11,8 +11,8 @@
 import ConfigSpace
 import flaml.tune
 import flaml.tune.sample
-from flaml.tune.sample import Domain
 import numpy as np
+from flaml.tune.sample import Domain
 
 if TYPE_CHECKING:
     from ConfigSpace.hyperparameters import Hyperparameter

mlos_core/mlos_core/tests/spaces/adapters/space_adapter_factory_test.py

@@ -29,7 +29,9 @@
     ],
 )
 def test_concrete_space_adapter_type(space_adapter_type: SpaceAdapterType) -> None:
-    """Test that all spaceadapter types are listed in the ConcreteSpaceAdapter constraints."""
+    """Test that all spaceadapter types are listed in the ConcreteSpaceAdapter
+    constraints.
+    """
     assert space_adapter_type.value in ConcreteSpaceAdapter.__args__