fixed mypy

optuna · May 26, 2022 · b0ff609 · b0ff609
1 parent 6b06b05
commit b0ff609
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 11 deletions.
diff --git a/benchmarks/kurobako/__init__.py b/benchmarks/kurobako/__init__.py
diff --git a/benchmarks/kurobako/problems/wfg/transformation_functions.py b/benchmarks/kurobako/problems/wfg/transformation_functions.py
@@ -5,15 +5,8 @@
 import numpy as np
 
 
-# class BaseTransformations(object, metaclass=abc.ABCMeta):
-#     @abc.abstractmethod
-#     def __call__(self, *args, **kwargs):
-
-#         raise NotImplementedError
-
-
-class BaseIdenticalTransformations(metaclass=abc.ABCMeta):
-    @abc.abstractclassmethod
+class BaseIdenticalTransformation(metaclass=abc.ABCMeta):
+    @abc.abstractmethod
     def __call__(self, y: float) -> float:
         raise NotImplementedError
 
@@ -40,14 +33,14 @@ def __call__(self, y: np.ndarray) -> float:
 
 
 BaseTransformations = Union[
-    BaseIdenticalTransformations,
+    BaseIdenticalTransformation,
     BaseBiasTransformation,
     BaseShiftTransformation,
     BaseReductionTransformation,
 ]
 
 
-class IdenticalTransformation(BaseIdenticalTransformations):
+class IdenticalTransformation(BaseIdenticalTransformation):
     def __init__(self) -> None:
         pass
 

diff --git a/benchmarks/naslib/__init__.py b/benchmarks/naslib/__init__.py