apply formats.sh

optuna_integration/_imports.py

@@ -1,4 +1,5 @@
 from __future__ import annotations
+
 import importlib
 import types
 from types import TracebackType

optuna_integration/_lightgbm_tuner/sklearn.py

@@ -1,4 +1,5 @@
 from __future__ import annotations
+
 from typing import Any
 import warnings
 

optuna_integration/allennlp/_dump_best_config.py

@@ -1,4 +1,5 @@
 from __future__ import annotations
+
 import json
 
 import optuna

optuna_integration/allennlp/_variables.py

@@ -1,4 +1,5 @@
 from __future__ import annotations
+
 import json
 import os
 from typing import Any

optuna_integration/botorch/botorch.py

@@ -1,8 +1,8 @@
 from __future__ import annotations
 
-from typing import Any
 from collections.abc import Callable
 from collections.abc import Sequence
+from typing import Any
 import warnings
 
 import numpy
@@ -974,15 +974,15 @@ def infer_relative_search_space(
         self,
         study: Study,
         trial: FrozenTrial,
-    ) -> Dict[str, BaseDistribution]:
+    ) -> dict[str, BaseDistribution]:
         if self._study_id is None:
             self._study_id = study._study_id
         if self._study_id != study._study_id:
             # Note that the check below is meaningless when `InMemoryStorage` is used
             # because `InMemoryStorage.create_new_study` always returns the same study ID.
             raise RuntimeError("BoTorchSampler cannot handle multiple studies.")
 
-        search_space: Dict[str, BaseDistribution] = {}
+        search_space: dict[str, BaseDistribution] = {}
         for name, distribution in self._search_space.calculate(study).items():
             if distribution.single():
                 # built-in `candidates_func` cannot handle distributions that contain just a
@@ -997,8 +997,8 @@ def sample_relative(
         self,
         study: Study,
         trial: FrozenTrial,
-        search_space: Dict[str, BaseDistribution],
-    ) -> Dict[str, Any]:
+        search_space: dict[str, BaseDistribution],
+    ) -> dict[str, Any]:
         assert isinstance(search_space, dict)
 
         if len(search_space) == 0:

optuna_integration/cma/cma.py

@@ -1,10 +1,10 @@
-
 from __future__ import annotations
+
+from collections.abc import Container
+from collections.abc import Sequence
 import math
 import random
 from typing import Any
-from collections.abc import Container
-from collections.abc import Sequence
 
 import numpy
 import optuna

optuna_integration/fastaiv2/fastaiv2.py

@@ -1,4 +1,5 @@
 from __future__ import annotations
+
 import optuna
 from packaging import version
 

optuna_integration/tfkeras/tfkeras.py

@@ -1,4 +1,5 @@
 from __future__ import annotations
+
 from typing import Any
 import warnings