additional test for config resolution and override

This commit is contained in:
Shital Shah 2020-05-03 04:41:51 -07:00
Родитель 6be9f214ab
Коммит 1d42b62d4f
1 изменённых файлов: 13 добавлений и 2 удалений

Просмотреть файл

@ -1,6 +1,6 @@
from archai.common.config import Config from archai.common.config import Config
def test_param_override(): def test_param_override1():
conf = Config('confs/algos/darts.yaml;confs/datasets/cifar10.yaml') conf = Config('confs/algos/darts.yaml;confs/datasets/cifar10.yaml')
assert not conf['nas']['eval']['trainer']['apex']['enabled'] assert not conf['nas']['eval']['trainer']['apex']['enabled']
assert not conf['nas']['eval']['loader']['apex']['enabled'] assert not conf['nas']['eval']['loader']['apex']['enabled']
@ -10,4 +10,15 @@ def test_param_override():
assert conf['nas']['eval']['trainer']['apex']['enabled'] assert conf['nas']['eval']['trainer']['apex']['enabled']
assert conf['nas']['eval']['loader']['apex']['enabled'] assert conf['nas']['eval']['loader']['apex']['enabled']
test_param_override() def test_param_override2():
conf = Config('confs/algos/darts.yaml;confs/datasets/imagenet.yaml')
assert conf['nas']['eval']['trainer']['apex']['distributed_enabled']
assert conf['nas']['eval']['loader']['apex']['distributed_enabled']
conf = Config('confs/algos/darts.yaml;confs/datasets/imagenet.yaml',
param_args=["--nas.eval.trainer.apex.enabled", "True", "--nas.eval.trainer.apex.distributed_enabled", "False"])
assert not conf['nas']['eval']['trainer']['apex']['distributed_enabled']
assert not conf['nas']['eval']['loader']['apex']['distributed_enabled']
test_param_override1()
test_param_override2()