update formatting with black v24.2.0

examples/instantiate/object/my_app.py

@@ -6,8 +6,7 @@
 
 
 class DBConnection:
-    def connect(self) -> None:
-        ...
+    def connect(self) -> None: ...
 
 
 class MySQLConnection(DBConnection):

examples/plugins/example_sweeper_plugin/tests/test_example_sweeper_plugin.py

@@ -52,8 +52,7 @@
         )
     ],
 )
-class TestExampleSweeper(LauncherTestSuite):
-    ...
+class TestExampleSweeper(LauncherTestSuite): ...
 
 
 # Many sweepers are batching jobs in groups.
@@ -73,8 +72,7 @@
         )
     ],
 )
-class TestExampleSweeperWithBatching(BatchedSweeperTestSuite):
-    ...
+class TestExampleSweeperWithBatching(BatchedSweeperTestSuite): ...
 
 
 # Run integration test suite with the basic launcher and this sweeper

hydra/_internal/config_repository.py

@@ -27,34 +27,27 @@
 
 class IConfigRepository(ABC):
     @abstractmethod
-    def get_schema_source(self) -> ConfigSource:
-        ...
+    def get_schema_source(self) -> ConfigSource: ...
 
     @abstractmethod
-    def load_config(self, config_path: str) -> Optional[ConfigResult]:
-        ...
+    def load_config(self, config_path: str) -> Optional[ConfigResult]: ...
 
     @abstractmethod
-    def group_exists(self, config_path: str) -> bool:
-        ...
+    def group_exists(self, config_path: str) -> bool: ...
 
     @abstractmethod
-    def config_exists(self, config_path: str) -> bool:
-        ...
+    def config_exists(self, config_path: str) -> bool: ...
 
     @abstractmethod
     def get_group_options(
         self, group_name: str, results_filter: Optional[ObjectType] = ObjectType.CONFIG
-    ) -> List[str]:
-        ...
+    ) -> List[str]: ...
 
     @abstractmethod
-    def get_sources(self) -> List[ConfigSource]:
-        ...
+    def get_sources(self) -> List[ConfigSource]: ...
 
     @abstractmethod
-    def initialize_sources(self, config_search_path: ConfigSearchPath) -> None:
-        ...
+    def initialize_sources(self, config_search_path: ConfigSearchPath) -> None: ...
 
 
 class ConfigRepository(IConfigRepository):

hydra/core/config_loader.py

@@ -23,26 +23,21 @@
         run_mode: RunMode,
         from_shell: bool = True,
         validate_sweep_overrides: bool = True,
-    ) -> DictConfig:
-        ...
+    ) -> DictConfig: ...
 
     @abstractmethod
     def load_sweep_config(
         self, master_config: DictConfig, sweep_overrides: List[str]
-    ) -> DictConfig:
-        ...
+    ) -> DictConfig: ...
 
     @abstractmethod
-    def get_search_path(self) -> ConfigSearchPath:
-        ...
+    def get_search_path(self) -> ConfigSearchPath: ...
 
     @abstractmethod
-    def get_sources(self) -> List[ConfigSource]:
-        ...
+    def get_sources(self) -> List[ConfigSource]: ...
 
     @abstractmethod
-    def list_groups(self, parent_name: str) -> List[str]:
-        ...
+    def list_groups(self, parent_name: str) -> List[str]: ...
 
     @abstractmethod
     def get_group_options(
@@ -51,14 +46,12 @@
         results_filter: Optional[ObjectType] = ObjectType.CONFIG,
         config_name: Optional[str] = None,
         overrides: Optional[List[str]] = None,
-    ) -> List[str]:
-        ...
+    ) -> List[str]: ...
 
     @abstractmethod
     def compute_defaults_list(
         self,
         config_name: Optional[str],
         overrides: List[str],
         run_mode: RunMode,
-    ) -> Any:
-        ...
+    ) -> Any: ...

hydra/core/config_search_path.py

@@ -28,8 +28,7 @@
 
 class ConfigSearchPath(ABC):
     @abstractmethod
-    def get_path(self) -> MutableSequence[SearchPathElement]:
-        ...
+    def get_path(self) -> MutableSequence[SearchPathElement]: ...
 
     @abstractmethod
     def append(

hydra/core/config_store.py

@@ -28,8 +28,7 @@
             group=group, name=name, node=node, package=package, provider=self.provider
         )
 
-    def __exit__(self, exc_type: Any, exc_value: Any, exc_traceback: Any) -> Any:
-        ...
+    def __exit__(self, exc_type: Any, exc_value: Any, exc_traceback: Any) -> Any: ...
 
 
 @dataclass

hydra/errors.py

@@ -2,12 +2,10 @@
 from typing import Optional, Sequence
 
 
-class HydraException(Exception):
-    ...
+class HydraException(Exception): ...
 
 
-class CompactHydraException(HydraException):
-    ...
+class CompactHydraException(HydraException): ...
 
 
 class OverrideParseException(CompactHydraException):
@@ -17,16 +15,13 @@
         self.message = message
 
 
-class InstantiationException(CompactHydraException):
-    ...
+class InstantiationException(CompactHydraException): ...
 
 
-class ConfigCompositionException(CompactHydraException):
-    ...
+class ConfigCompositionException(CompactHydraException): ...
 
 
-class SearchPathException(CompactHydraException):
-    ...
+class SearchPathException(CompactHydraException): ...
 
 
 class MissingConfigException(IOError, ConfigCompositionException):
@@ -41,5 +36,4 @@
         self.options = options
 
 
-class HydraDeprecationError(HydraException):
-    ...
+class HydraDeprecationError(HydraException): ...

hydra/extra/pytest_plugin.py

@@ -21,21 +21,19 @@ def hydra_restore_singletons() -> Generator[None, None, None]:
 
 
 @fixture(scope="function")
-def hydra_sweep_runner() -> (
-    Callable[
-        [
-            Optional[str],
-            Optional[str],
-            Optional[TaskFunction],
-            Optional[str],
-            Optional[str],
-            Optional[List[str]],
-            Optional[Path],
-            bool,
-        ],
-        SweepTaskFunction,
-    ]
-):
+def hydra_sweep_runner() -> Callable[
+    [
+        Optional[str],
+        Optional[str],
+        Optional[TaskFunction],
+        Optional[str],
+        Optional[str],
+        Optional[List[str]],
+        Optional[Path],
+        bool,
+    ],
+    SweepTaskFunction,
+]:
     def _(
         calling_file: Optional[str],
         calling_module: Optional[str],
@@ -61,19 +59,17 @@ def _(
 
 
 @fixture(scope="function")
-def hydra_task_runner() -> (
-    Callable[
-        [
-            Optional[str],
-            Optional[str],
-            Optional[str],
-            Optional[str],
-            Optional[List[str]],
-            bool,
-        ],
-        TaskTestFunction,
-    ]
-):
+def hydra_task_runner() -> Callable[
+    [
+        Optional[str],
+        Optional[str],
+        Optional[str],
+        Optional[str],
+        Optional[List[str]],
+        bool,
+    ],
+    TaskTestFunction,
+]:
     def _(
         calling_file: Optional[str],
         calling_module: Optional[str],

hydra/initialize.py

@@ -95,8 +95,7 @@
             job_name=job_name,
         )
 
-    def __enter__(self, *args: Any, **kwargs: Any) -> None:
-        ...
+    def __enter__(self, *args: Any, **kwargs: Any) -> None: ...
 
     def __exit__(self, exc_type: Any, exc_val: Any, exc_tb: Any) -> None:
         restore_gh_from_backup(self._gh_backup)
@@ -130,8 +129,7 @@
             job_name=job_name,
         )
 
-    def __enter__(self, *args: Any, **kwargs: Any) -> None:
-        ...
+    def __enter__(self, *args: Any, **kwargs: Any) -> None: ...
 
     def __exit__(self, exc_type: Any, exc_val: Any, exc_tb: Any) -> None:
         restore_gh_from_backup(self._gh_backup)
@@ -169,8 +167,7 @@
         csp = create_config_search_path(search_path_dir=config_dir)
         Hydra.create_main_hydra2(task_name=job_name, config_search_path=csp)
 
-    def __enter__(self, *args: Any, **kwargs: Any) -> None:
-        ...
+    def __enter__(self, *args: Any, **kwargs: Any) -> None: ...
 
     def __exit__(self, exc_type: Any, exc_val: Any, exc_tb: Any) -> None:
         restore_gh_from_backup(self._gh_backup)

hydra/plugins/completion_plugin.py

@@ -31,12 +31,10 @@
         self.config_loader = config_loader
 
     @abstractmethod
-    def install(self) -> None:
-        ...
+    def install(self) -> None: ...
 
     @abstractmethod
-    def uninstall(self) -> None:
-        ...
+    def uninstall(self) -> None: ...
 
     @staticmethod
     @abstractmethod
@@ -47,8 +45,7 @@
         ...
 
     @abstractmethod
-    def query(self, config_name: Optional[str]) -> None:
-        ...
+    def query(self, config_name: Optional[str]) -> None: ...
 
     @staticmethod
     @abstractmethod

hydra/plugins/config_source.py

@@ -47,20 +47,17 @@
         ...
 
     @abstractmethod
-    def load_config(self, config_path: str) -> ConfigResult:
-        ...
+    def load_config(self, config_path: str) -> ConfigResult: ...
 
     # subclasses may override to improve performance
     def exists(self, config_path: str) -> bool:
         return self.is_group(config_path) or self.is_config(config_path)
 
     @abstractmethod
-    def is_group(self, config_path: str) -> bool:
-        ...
+    def is_group(self, config_path: str) -> bool: ...
 
     @abstractmethod
-    def is_config(self, config_path: str) -> bool:
-        ...
+    def is_config(self, config_path: str) -> bool: ...
 
     @abstractmethod
     def available(self) -> bool:

hydra/plugins/plugin.py

@@ -2,5 +2,4 @@
 from abc import ABC
 
 
-class Plugin(ABC):
-    ...
+class Plugin(ABC): ...

hydra/plugins/search_path_plugin.py

@@ -8,5 +8,4 @@
 
 class SearchPathPlugin(Plugin):
     @abstractmethod
-    def manipulate_search_path(self, search_path: ConfigSearchPath) -> None:
-        ...
+    def manipulate_search_path(self, search_path: ConfigSearchPath) -> None: ...

hydra/test_utils/test_utils.py

@@ -102,8 +102,7 @@
         config_name: Optional[str],
         overrides: Optional[List[str]] = None,
         configure_logging: bool = False,
-    ) -> TaskTestFunction:
-        ...
+    ) -> TaskTestFunction: ...
 
 
 class SweepTaskFunction:
@@ -184,8 +183,7 @@
         config_name: Optional[str],
         overrides: Optional[List[str]],
         temp_dir: Optional[Path] = None,
-    ) -> SweepTaskFunction:
-        ...
+    ) -> SweepTaskFunction: ...
 
 
 def chdir_hydra_root(subdir: Optional[str] = None) -> None:

plugins/hydra_nevergrad_sweeper/hydra_plugins/hydra_nevergrad_sweeper/_impl.py

@@ -129,9 +129,9 @@ def sweep(self, arguments: List[str]) -> None:
         parsed = parser.parse_overrides(arguments)
 
         for override in parsed:
-            params[
-                override.get_key_element()
-            ] = create_nevergrad_parameter_from_override(override)
+            params[override.get_key_element()] = (
+                create_nevergrad_parameter_from_override(override)
+            )
 
         parametrization = ng.p.Dict(**params)
         parametrization.function.deterministic = not self.opt_config.noisy

plugins/hydra_ray_launcher/tests/test_ray_aws_launcher.py

@@ -329,7 +329,6 @@ class TestRayAWSLauncher(LauncherTestSuite):
     ],
 )
 class TestRayAWSLauncherIntegration(IntegrationTestSuite):
-
     """
     Run this launcher through the integration test suite.
     """