add explicit class init-arguments

SimonBlanke · Apr 29, 2024 · abf012d · abf012d
1 parent a9df628
commit abf012d
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 6 deletions.
diff --git a/src/surfaces/test_functions/_base_test_function.py b/src/surfaces/test_functions/_base_test_function.py
@@ -20,7 +20,7 @@ class BaseTestFunction:
     objective_function: callable
     pure_objective_function: callable
 
-    def __init__(self, metric, sleep=0, evaluate_from_data=False):
+    def __init__(self, metric, sleep, evaluate_from_data):
         self.sleep = sleep
         self.metric = metric
 

diff --git a/src/surfaces/test_functions/machine_learning/tabular_classifiers.py b/src/surfaces/test_functions/machine_learning/tabular_classifiers.py
@@ -20,8 +20,13 @@ class KNeighborsClassifierFunction(MachineLearningFunction):
     cv_default = [2, 3, 4, 5, 8, 10]
     dataset_default = [digits_data, wine_data, iris_data]
 
-    def __init__(self, *args, **kwargs):
-        super().__init__(*args, **kwargs)
+    def __init__(
+        self,
+        metric="loss",
+        sleep=0,
+        evaluate_from_data=False,
+    ):
+        super().__init__(metric, sleep, evaluate_from_data)
 
     def search_space(
         self,

diff --git a/src/surfaces/test_functions/mathematical/_base_objective_function.py b/src/surfaces/test_functions/mathematical/_base_objective_function.py
@@ -17,13 +17,12 @@ class MathematicalFunction(BaseTestFunction):
 
     def __init__(
         self,
-        *args,
         metric="loss",
         input_type="dictionary",
         sleep=0,
-        **kwargs,
+        evaluate_from_data=False,
     ):
-        super().__init__(*args, metric, **kwargs)
+        super().__init__(metric, sleep, evaluate_from_data)
 
         self.metric = metric
         self.input_type = input_type