Skip to content

Commit b96bb07

Browse files
committed
move method to private
1 parent 50c3f7f commit b96bb07

30 files changed

+63
-63
lines changed

src/surfaces/test_functions/_base_test_function.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -43,10 +43,10 @@ class BaseTestFunction:
4343
default_bounds: Tuple[float, float] = (-5.0, 5.0)
4444

4545
def _create_objective_function_(func):
46-
"""Decorator that calls create_objective_function after __init__."""
46+
"""Decorator that calls _create_objective_function after __init__."""
4747
def wrapper(self, *args, **kwargs):
4848
func(self, *args, **kwargs)
49-
self.create_objective_function()
49+
self._create_objective_function()
5050
return wrapper
5151

5252
@_create_objective_function_
@@ -55,15 +55,15 @@ def __init__(self, metric, sleep, validate=True):
5555
self.metric = metric
5656
self._validate = validate
5757

58-
def create_objective_function(self):
59-
raise NotImplementedError("'create_objective_function' must be implemented")
58+
def _create_objective_function(self):
59+
raise NotImplementedError("'_create_objective_function' must be implemented")
6060

6161
@property
6262
def default_search_space(self) -> Dict[str, Any]:
6363
"""Default search space for this function (override in subclasses)."""
6464
raise NotImplementedError("'default_search_space' must be implemented")
6565

66-
def return_metric(self, value):
66+
def _return_metric(self, value):
6767
"""Transform raw value based on metric setting (override in subclasses)."""
6868
return value
6969

@@ -110,7 +110,7 @@ def _evaluate_with_timing(self, params: Dict[str, Any]) -> float:
110110
"""Evaluate with sleep timing applied."""
111111
time.sleep(self.sleep)
112112
raw_value = self.pure_objective_function(params)
113-
return self.return_metric(raw_value)
113+
return self._return_metric(raw_value)
114114

115115
def _evaluate_raw(self, params: Dict[str, Any]) -> float:
116116
"""Evaluate without metric transformation (internal use)."""

src/surfaces/test_functions/machine_learning/_base_machine_learning.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -74,12 +74,12 @@ def _evaluate_with_timing(self, params: Dict[str, Any]) -> float:
7474
"""Evaluate with sleep timing applied (overrides base class)."""
7575
time.sleep(self.sleep)
7676
raw_value = self._get_raw_value(params)
77-
return self.return_metric(raw_value)
77+
return self._return_metric(raw_value)
7878

7979
# Metrics that behave like scores (higher is better)
8080
SCORE_LIKE_METRICS = {"score", "accuracy", "r2", "f1", "precision", "recall", "auc"}
8181

82-
def return_metric(self, score: float) -> float:
82+
def _return_metric(self, score: float) -> float:
8383
"""
8484
Transform raw score value based on metric setting.
8585

src/surfaces/test_functions/machine_learning/tabular/classification/test_functions/k_neighbors_classifier.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -81,7 +81,7 @@ def _search_space(
8181

8282
return search_space
8383

84-
def create_objective_function(self):
84+
def _create_objective_function(self):
8585
def k_neighbors_classifier(params):
8686
knc = KNeighborsClassifier(
8787
n_neighbors=params["n_neighbors"],

src/surfaces/test_functions/machine_learning/tabular/regression/test_functions/gradient_boosting_regressor.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -82,7 +82,7 @@ def _search_space(
8282

8383
return search_space
8484

85-
def create_objective_function(self):
85+
def _create_objective_function(self):
8686
def gradient_boosting_regressor(params):
8787
knc = GradientBoostingRegressor(
8888
n_estimators=params["n_estimators"],

src/surfaces/test_functions/machine_learning/tabular/regression/test_functions/k_neighbors_regressor.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -82,7 +82,7 @@ def _search_space(
8282

8383
return search_space
8484

85-
def create_objective_function(self):
85+
def _create_objective_function(self):
8686
def k_neighbors_regressor(params):
8787
knc = KNeighborsRegressor(
8888
n_neighbors=params["n_neighbors"],

src/surfaces/test_functions/mathematical/_base_mathematical_function.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -62,7 +62,7 @@ def _create_search_space(
6262
Internal method for creating search spaces with non-default parameters.
6363
For most use cases, access default_search_space property instead.
6464
"""
65-
return self.create_n_dim_search_space(min=min, max=max, size=size, value_types=value_types)
65+
return self._create_n_dim_search_space(min=min, max=max, size=size, value_types=value_types)
6666

6767
def __init__(
6868
self,
@@ -85,7 +85,7 @@ def __init__(
8585
self.metric = metric
8686
self.sleep = sleep
8787

88-
def return_metric(self, loss: float) -> float:
88+
def _return_metric(self, loss: float) -> float:
8989
"""
9090
Transform raw loss value based on metric setting.
9191
@@ -118,14 +118,14 @@ def _to_score(self, raw_value: float) -> float:
118118
return -raw_value
119119

120120
@staticmethod
121-
def conv_arrays2lists(search_space: Dict[str, Any]) -> Dict[str, list]:
121+
def _conv_arrays2lists(search_space: Dict[str, Any]) -> Dict[str, list]:
122122
"""Convert array-valued search space to list-valued."""
123123
return {
124124
para_name: list(dim_values)
125125
for para_name, dim_values in search_space.items()
126126
}
127127

128-
def create_n_dim_search_space(
128+
def _create_n_dim_search_space(
129129
self,
130130
min: Union[float, list] = -5,
131131
max: Union[float, list] = 5,

src/surfaces/test_functions/mathematical/test_functions_1d/gramacy_and_lee_function.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -61,7 +61,7 @@ def __init__(self, metric="score", sleep=0):
6161
super().__init__(metric, sleep)
6262
self.n_dim = 1
6363

64-
def create_objective_function(self):
64+
def _create_objective_function(self):
6565
def gramacy_and_lee_function(params):
6666
x = params["x0"]
6767

@@ -70,6 +70,6 @@ def gramacy_and_lee_function(params):
7070
self.pure_objective_function = gramacy_and_lee_function
7171

7272
def _search_space(self, min=0.5, max=2.5, value_types="array", size=10000):
73-
return super().create_n_dim_search_space(
73+
return super()._create_n_dim_search_space(
7474
min=min, max=max, size=size, value_types=value_types
7575
)

src/surfaces/test_functions/mathematical/test_functions_2d/ackley_function.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -77,7 +77,7 @@ def __init__(
7777
self.A = A
7878
self.angle = angle
7979

80-
def create_objective_function(self):
80+
def _create_objective_function(self):
8181
def ackley_function(params):
8282
x = params["x0"]
8383
y = params["x1"]
@@ -92,6 +92,6 @@ def ackley_function(params):
9292
self.pure_objective_function = ackley_function
9393

9494
def _search_space(self, min=-5, max=5, value_types="array", size=10000):
95-
return super().create_n_dim_search_space(
95+
return super()._create_n_dim_search_space(
9696
min=min, max=max, size=size, value_types=value_types
9797
)

src/surfaces/test_functions/mathematical/test_functions_2d/beale_function.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -70,7 +70,7 @@ def __init__(self, A=1.5, B=2.25, C=2.652, metric="score", sleep=0):
7070
self.B = B
7171
self.C = C
7272

73-
def create_objective_function(self):
73+
def _create_objective_function(self):
7474
def beale_function(params):
7575
x = params["x0"]
7676
y = params["x1"]
@@ -84,6 +84,6 @@ def beale_function(params):
8484
self.pure_objective_function = beale_function
8585

8686
def _search_space(self, min=-4.5, max=4.5, value_types="array", size=10000):
87-
return super().create_n_dim_search_space(
87+
return super()._create_n_dim_search_space(
8888
min=min, max=max, size=size, value_types=value_types
8989
)

src/surfaces/test_functions/mathematical/test_functions_2d/booth_function.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -55,7 +55,7 @@ def __init__(self, metric="score", sleep=0):
5555
super().__init__(metric, sleep)
5656
self.n_dim = 2
5757

58-
def create_objective_function(self):
58+
def _create_objective_function(self):
5959

6060
def booth_function(params):
6161
x = params["x0"]
@@ -69,6 +69,6 @@ def booth_function(params):
6969
self.pure_objective_function = booth_function
7070

7171
def _search_space(self, min=-10, max=10, value_types="array", size=10000):
72-
return super().create_n_dim_search_space(
72+
return super()._create_n_dim_search_space(
7373
min=min, max=max, size=size, value_types=value_types
7474
)

0 commit comments

Comments
 (0)