[dev] Config Registration Functions in __init__.py

hydra-configs-torch/hydra_configs/torch/optim/__init__.py

@@ -1,3 +1,4 @@
+# Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved
 # flake8: noqa
 # Mirrors torch/optim __init__ to allow for symmetric import structure
 from .adadelta import AdadeltaConf
@@ -14,6 +15,25 @@
 from .lbfgs import LBFGSConf
 from . import lr_scheduler
 
+from hydra.core.config_store import ConfigStore
+
+
+def register_configs():
+    cs = ConfigStore.instance()
+    cs.store(provider="torch", group="torch/optim", name="adadelta", node=AdadeltaConf)
+    cs.store(provider="torch", group="torch/optim", name="adagrad", node=AdagradConf)
+    cs.store(provider="torch", group="torch/optim", name="adam", node=AdamConf)
+    cs.store(provider="torch", group="torch/optim", name="adamw", node=AdamWConf)
+    cs.store(
+        provider="torch", group="torch/optim", name="sparseadam", node=SparseAdamConf
+    )
+    cs.store(provider="torch", group="torch/optim", name="adamax", node=AdamaxConf)
+    cs.store(provider="torch", group="torch/optim", name="asgd", node=ASGDConf)
+    cs.store(provider="torch", group="torch/optim", name="sgd", node=SGDConf)
+    cs.store(provider="torch", group="torch/optim", name="lbfgs", node=LBFGSConf)
+    cs.store(provider="torch", group="torch/optim", name="rprop", node=RpropConf)
+    cs.store(provider="torch", group="torch/optim", name="rmsprop", node=RMSpropConf)
+
 
 del adadelta
 del adagrad

hydra-configs-torch/tests/test_register_optimizers.py

@@ -0,0 +1,29 @@
+# Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved
+from hydra.core.config_store import ConfigStore
+import hydra_configs.torch.optim
+
+cs = ConfigStore()
+
+# registers all 'base' optimizer configs with configstore instance
+hydra_configs.torch.optim.register()
+
+expected = set(
+    [
+        "adadelta",
+        "adagrad",
+        "adam",
+        "adamw",
+        "sparseadam",
+        "adamax",
+        "asgd",
+        "sgd",
+        "lbfgs",
+        "rprop",
+        "rmsprop",
+    ]
+)
+
+
+def test_instantiate_classes() -> None:
+    actual = set([conf.split(".yaml")[0] for conf in cs.list("torch/optim")])
+    assert not actual ^ expected