@@ -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 .mo_benchmark = NASBench1shot1SearchSpace1Benchmark (data_path = data_path , rng = rng )
601+ self .mo_benchmark = NASBench1shot1SearchSpace1MOBenchmark (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 .mo_benchmark = NASBench1shot1SearchSpace2Benchmark (data_path = data_path , rng = rng )
612+ self .mo_benchmark = NASBench1shot1SearchSpace2MOBenchmark (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 .mo_benchmark = NASBench1shot1SearchSpace3Benchmark (data_path = data_path , rng = rng )
623+ self .mo_benchmark = NASBench1shot1SearchSpace3MOBenchmark (data_path = data_path , rng = rng )
624624
625625 @staticmethod
626626 def get_configuration_space (seed : Union [int , None ] = None ) -> CS .ConfigurationSpace :
0 commit comments