Skip to content

Commit c2f1454

Browse files
committed
NAS1shot1: Solve SO Inheritance Problem
1 parent 0b6d097 commit c2f1454

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

hpobench/benchmarks/nas/nasbench_1shot1.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)