@@ -41,14 +41,6 @@ def __init__(self, hparams):
4141 self .save_hyperparameters (hparams )
4242
4343
44- class AssignHparamsModel (BoringModel ):
45- """ Tests that a model can take an object with explicit setter """
46-
47- def __init__ (self , hparams ):
48- super ().__init__ ()
49- self .hparams = hparams
50-
51-
5244def decorate (func ):
5345
5446 @functools .wraps (func )
@@ -68,16 +60,6 @@ def __init__(self, hparams, *my_args, **my_kwargs):
6860 self .save_hyperparameters (hparams )
6961
7062
71- class AssignHparamsDecoratedModel (BoringModel ):
72- """ Tests that a model can take an object with explicit setter"""
73-
74- @decorate
75- @decorate
76- def __init__ (self , hparams , * my_args , ** my_kwargs ):
77- super ().__init__ ()
78- self .hparams = hparams
79-
80-
8163# -------------------------
8264# STANDARD TESTS
8365# -------------------------
@@ -114,7 +96,7 @@ def _run_standard_hparams_test(tmpdir, model, cls, try_overwrite=False):
11496
11597
11698@pytest .mark .parametrize (
117- "cls" , [SaveHparamsModel , AssignHparamsModel , SaveHparamsDecoratedModel , AssignHparamsDecoratedModel ]
99+ "cls" , [SaveHparamsModel , SaveHparamsDecoratedModel ]
118100)
119101def test_namespace_hparams (tmpdir , cls ):
120102 # init model
@@ -125,7 +107,7 @@ def test_namespace_hparams(tmpdir, cls):
125107
126108
127109@pytest .mark .parametrize (
128- "cls" , [SaveHparamsModel , AssignHparamsModel , SaveHparamsDecoratedModel , AssignHparamsDecoratedModel ]
110+ "cls" , [SaveHparamsModel , SaveHparamsDecoratedModel ]
129111)
130112def test_dict_hparams (tmpdir , cls ):
131113 # init model
@@ -136,7 +118,7 @@ def test_dict_hparams(tmpdir, cls):
136118
137119
138120@pytest .mark .parametrize (
139- "cls" , [SaveHparamsModel , AssignHparamsModel , SaveHparamsDecoratedModel , AssignHparamsDecoratedModel ]
121+ "cls" , [SaveHparamsModel , SaveHparamsDecoratedModel ]
140122)
141123def test_omega_conf_hparams (tmpdir , cls ):
142124 # init model
@@ -580,8 +562,7 @@ class SuperClassPositionalArgs(BoringModel):
580562
581563 def __init__ (self , hparams ):
582564 super ().__init__ ()
583- self ._hparams = None # pretend BoringModel did not call self.save_hyperparameters()
584- self .hparams = hparams
565+ self ._hparams = hparams # pretend BoringModel did not call self.save_hyperparameters()
585566
586567
587568class SubClassVarArgs (SuperClassPositionalArgs ):
@@ -617,8 +598,6 @@ def test_init_arg_with_runtime_change(tmpdir, cls):
617598 assert model .hparams .running_arg == 123
618599 model .hparams .running_arg = - 1
619600 assert model .hparams .running_arg == - 1
620- model .hparams = Namespace (abc = 42 )
621- assert model .hparams .abc == 42
622601
623602 trainer = Trainer (
624603 default_root_dir = tmpdir ,
@@ -664,18 +643,11 @@ class TestHydraModel(BoringModel):
664643
665644 def __init__ (self , args_0 , args_1 , args_2 , kwarg_1 = None ):
666645 self .save_hyperparameters ()
667- self .test_hparams ()
668- config_file = f"{ tmpdir } /hparams.yaml"
669- save_hparams_to_yaml (config_file , self .hparams )
670- self .hparams = load_hparams_from_yaml (config_file )
671- self .test_hparams ()
672- super ().__init__ ()
673-
674- def test_hparams (self ):
675646 assert self .hparams .args_0 .log == "Something"
676647 assert self .hparams .args_1 ['cfg' ].log == "Something"
677648 assert self .hparams .args_2 [0 ].log == "Something"
678649 assert self .hparams .kwarg_1 ['cfg' ][0 ].log == "Something"
650+ super ().__init__ ()
679651
680652 with initialize (config_path = "conf" ):
681653 args_0 = compose (config_name = "config" )
0 commit comments