Spaces:
Running
Running
| # Copyright (c) Facebook, Inc. and its affiliates. | |
| import os | |
| import tempfile | |
| import unittest | |
| import yaml | |
| from omegaconf import OmegaConf | |
| from omegaconf import __version__ as oc_version | |
| from dataclasses import dataclass | |
| from detectron2.config import instantiate, LazyCall as L | |
| from detectron2.layers import ShapeSpec | |
| OC_VERSION = tuple(int(x) for x in oc_version.split(".")[:2]) | |
| class TestClass: | |
| def __init__(self, int_arg, list_arg=None, dict_arg=None, extra_arg=None): | |
| self.int_arg = int_arg | |
| self.list_arg = list_arg | |
| self.dict_arg = dict_arg | |
| self.extra_arg = extra_arg | |
| def __call__(self, call_arg): | |
| return call_arg + self.int_arg | |
| class TestDataClass: | |
| x: int | |
| y: str | |
| class TestConstruction(unittest.TestCase): | |
| def test_basic_construct(self): | |
| objconf = L(TestClass)( | |
| int_arg=3, | |
| list_arg=[10], | |
| dict_arg={}, | |
| extra_arg=L(TestClass)(int_arg=4, list_arg="${..list_arg}"), | |
| ) | |
| obj = instantiate(objconf) | |
| self.assertIsInstance(obj, TestClass) | |
| self.assertEqual(obj.int_arg, 3) | |
| self.assertEqual(obj.extra_arg.int_arg, 4) | |
| self.assertEqual(obj.extra_arg.list_arg, obj.list_arg) | |
| objconf.extra_arg.list_arg = [5] | |
| obj = instantiate(objconf) | |
| self.assertIsInstance(obj, TestClass) | |
| self.assertEqual(obj.extra_arg.list_arg, [5]) | |
| def test_instantiate_other_obj(self): | |
| # do nothing for other obj | |
| self.assertEqual(instantiate(5), 5) | |
| x = [3, 4, 5] | |
| self.assertEqual(instantiate(x), x) | |
| x = TestClass(1) | |
| self.assertIs(instantiate(x), x) | |
| x = {"xx": "yy"} | |
| self.assertIs(instantiate(x), x) | |
| def test_instantiate_lazy_target(self): | |
| # _target_ is result of instantiate | |
| objconf = L(L(len)(int_arg=3))(call_arg=4) | |
| objconf._target_._target_ = TestClass | |
| self.assertEqual(instantiate(objconf), 7) | |
| def test_instantiate_lst(self): | |
| lst = [1, 2, L(TestClass)(int_arg=1)] | |
| x = L(TestClass)(int_arg=lst) # list as an argument should be recursively instantiated | |
| x = instantiate(x).int_arg | |
| self.assertEqual(x[:2], [1, 2]) | |
| self.assertIsInstance(x[2], TestClass) | |
| self.assertEqual(x[2].int_arg, 1) | |
| def test_instantiate_namedtuple(self): | |
| x = L(TestClass)(int_arg=ShapeSpec(channels=1, width=3)) | |
| # test serialization | |
| with tempfile.TemporaryDirectory() as d: | |
| fname = os.path.join(d, "d2_test.yaml") | |
| OmegaConf.save(x, fname) | |
| with open(fname) as f: | |
| x = yaml.unsafe_load(f) | |
| x = instantiate(x) | |
| self.assertIsInstance(x.int_arg, ShapeSpec) | |
| self.assertEqual(x.int_arg.channels, 1) | |
| def test_bad_lazycall(self): | |
| with self.assertRaises(Exception): | |
| L(3) | |
| def test_instantiate_dataclass(self): | |
| a = L(TestDataClass)(x=1, y="s") | |
| a = instantiate(a) | |
| self.assertEqual(a.x, 1) | |
| self.assertEqual(a.y, "s") | |