mirror of
https://github.com/facebookresearch/pytorch3d.git
synced 2025-12-19 05:40:34 +08:00
small fixes to config
Summary: - indicate location of OmegaConf.structured failures - split the data gathering from enable_get_default_args to ease experimenting with it. - comment fixes. - nicer error when a_class_type has weird type. Reviewed By: kjchalup Differential Revision: D39434447 fbshipit-source-id: b80c7941547ca450e848038ef5be95b7ebbe8f3e
This commit is contained in:
committed by
Facebook GitHub Bot
parent
cb7bd33e7f
commit
da7fe2854e
@@ -168,7 +168,7 @@ class TestConfig(unittest.TestCase):
|
||||
self.assertIn(Banana, all_fruit)
|
||||
self.assertIn(Pear, all_fruit)
|
||||
self.assertIn(LargePear, all_fruit)
|
||||
self.assertEqual(set(registry.get_all(Pear)), {LargePear})
|
||||
self.assertEqual(registry.get_all(Pear), [LargePear])
|
||||
|
||||
@registry.register
|
||||
class Apple(Fruit):
|
||||
@@ -178,7 +178,7 @@ class TestConfig(unittest.TestCase):
|
||||
class CrabApple(Apple):
|
||||
pass
|
||||
|
||||
self.assertEqual(set(registry.get_all(Apple)), {CrabApple})
|
||||
self.assertEqual(registry.get_all(Apple), [CrabApple])
|
||||
|
||||
self.assertIs(registry.get(Fruit, "CrabApple"), CrabApple)
|
||||
|
||||
@@ -601,6 +601,7 @@ class TestConfig(unittest.TestCase):
|
||||
|
||||
for C_ in [C, C_fn, C_cl]:
|
||||
base = get_default_args(C_)
|
||||
self.assertEqual(OmegaConf.to_yaml(base), "a: B1\n")
|
||||
self.assertEqual(base.a, A.B1)
|
||||
replaced = OmegaConf.merge(base, {"a": "B2"})
|
||||
self.assertEqual(replaced.a, A.B2)
|
||||
|
||||
Reference in New Issue
Block a user