1111from multiprocessing import set_start_method , get_start_method
1212
1313# old Kriging with attribute "name" kriging
14- from spotpython .build .kriging import Kriging as OldKriging
14+ # from spotpython.build.kriging import Kriging as OldKriging
1515
1616# new Kriging without attribute "name" Kriging
1717from spotpython .surrogate .kriging import Kriging
@@ -114,7 +114,7 @@ class Spot:
114114 information about the optimizer stored as a dictionary.
115115 Default value is `optimizer_control_init()`.
116116 surrogate (object):
117- surrogate model. If `None`, spotpython's `kriging ` is used. Default value is `None`.
117+ surrogate model. If `None`, spotpython's `Kriging ` is used. Default value is `None`.
118118 surrogate_control (Dict[str, Union[int, float]]):
119119 surrogate model information stored as a dictionary.
120120 Default value is `surrogate_control_init()`.
@@ -411,7 +411,7 @@ def _surrogate_setup(self, surrogate) -> None:
411411 # Surrogate related information:
412412 self .surrogate = surrogate
413413 # If no surrogate model is specified, use the internal
414- # spotpython kriging surrogate:
414+ # spotpython Kriging surrogate:
415415 if self .surrogate is None :
416416 # Call kriging with surrogate_control parameters:
417417 self .surrogate = Kriging (
@@ -502,7 +502,7 @@ def get_spot_attributes_as_df(self) -> pd.DataFrame:
502502 32 show_models False
503503 33 show_progress True
504504 34 spot_writer None
505- 35 surrogate <spotpython.build .kriging.Kriging object at 0x...
505+ 35 surrogate <spotpython.surrogate .kriging.Kriging object at 0x...
506506 36 surrogate_control {'method': "regession", 'model_optimizer': <function ...
507507 37 tolerance_x 0
508508 38 upper [1]
@@ -1914,7 +1914,7 @@ def infill(self, x) -> float:
19141914 """
19151915 Infill (acquisition) function. Evaluates one point on the surrogate via `surrogate.predict(x.reshape(1,-1))`,
19161916 if `sklearn` surrogates are used or `surrogate.predict(x.reshape(1,-1), return_val=self.infill_criterion)`
1917- if the internal surrogate `kriging ` is selected.
1917+ if the internal surrogate `Kriging ` is selected.
19181918 This method is passed to the optimizer in `suggest_new_X`, i.e., the optimizer is called via
19191919 `self.optimizer(func=self.infill)`.
19201920
@@ -1931,7 +1931,7 @@ def infill(self, x) -> float:
19311931 """
19321932 # Reshape x to have shape (1, -1) because the predict method expects a 2D array
19331933 X = x .reshape (1 , - 1 )
1934- if isinstance (self .surrogate , Kriging ) and getattr (self .surrogate , "name" , None ) == "kriging " :
1934+ if isinstance (self .surrogate , Kriging ) and getattr (self .surrogate , "name" , None ) == "Kriging " :
19351935 return self .surrogate .predict (X , return_val = self .infill_criterion )
19361936 else :
19371937 return self .surrogate .predict (X )
0 commit comments