@@ -448,7 +448,7 @@ def __init__(self, data_path: Union[Path, str, None] = None,
448448 rng : np.random.RandomState, int, None
449449 Random seed for the benchmarks
450450 """
451- self .mo_benchmark = NASBench1shot1BaseMOBenchmark ( data_path = data_path , rng = rng )
451+ self .mo_benchmark : Union [ NASBench1shot1BaseMOBenchmark , None ] = None
452452 super (NASBench1shot1BaseBenchmark , self ).__init__ (rng = rng )
453453
454454 # pylint: disable=arguments-differ
@@ -598,7 +598,7 @@ class NASBench1shot1SearchSpace1Benchmark(NASBench1shot1BaseBenchmark):
598598 def __init__ (self , data_path : Union [Path , str , None ] = None ,
599599 rng : Union [np .random .RandomState , int , None ] = None ):
600600 super (NASBench1shot1SearchSpace1Benchmark , self ).__init__ (data_path = data_path , rng = rng )
601- self .search_space = SearchSpace1 ( )
601+ self .mo_benchmark = NASBench1shot1SearchSpace1Benchmark ( data_path = data_path , rng = rng )
602602
603603 @staticmethod
604604 def get_configuration_space (seed : Union [int , None ] = None ) -> CS .ConfigurationSpace :
@@ -609,7 +609,7 @@ class NASBench1shot1SearchSpace2Benchmark(NASBench1shot1BaseBenchmark):
609609 def __init__ (self , data_path : Union [Path , str , None ] = None ,
610610 rng : Union [np .random .RandomState , int , None ] = None ):
611611 super (NASBench1shot1SearchSpace2Benchmark , self ).__init__ (data_path = data_path , rng = rng )
612- self .search_space = SearchSpace2 ( )
612+ self .mo_benchmark = NASBench1shot1SearchSpace2Benchmark ( data_path = data_path , rng = rng )
613613
614614 @staticmethod
615615 def get_configuration_space (seed : Union [int , None ] = None ) -> CS .ConfigurationSpace :
@@ -620,7 +620,7 @@ class NASBench1shot1SearchSpace3Benchmark(NASBench1shot1BaseBenchmark):
620620 def __init__ (self , data_path : Union [Path , str , None ] = None ,
621621 rng : Union [np .random .RandomState , int , None ] = None ):
622622 super (NASBench1shot1SearchSpace3Benchmark , self ).__init__ (data_path = data_path , rng = rng )
623- self .search_space = SearchSpace3 ( )
623+ self .mo_benchmark = NASBench1shot1SearchSpace3Benchmark ( data_path = data_path , rng = rng )
624624
625625 @staticmethod
626626 def get_configuration_space (seed : Union [int , None ] = None ) -> CS .ConfigurationSpace :
0 commit comments