amend

benchmarl/__init__.py

@@ -22,6 +22,7 @@
     def _load_hydra_schemas():
         from hydra.core.config_store import ConfigStore
 
+        import benchmarl.environments
         from benchmarl.algorithms import algorithm_config_registry
         from benchmarl.environments import _task_class_registry
         from benchmarl.experiment import ExperimentConfig

benchmarl/environments/__init__.py

@@ -16,14 +16,14 @@
 
 # This is a registry mapping "envname/task_name" to the EnvNameTask.TASK_NAME enum
 # It is used by automatically load task enums from yaml files.
-# It is populated automatically.
+# It is populated automatically, do not modify.
 task_config_registry = {}
 
 # This is a registry mapping "envname_taskname" to the TaskConfig python dataclass of the task.
 # It is used by hydra to validate loaded configs.
 # You will see the "envname_taskname" strings in the hydra defaults at the top of yaml files.
 # This is optional and, if a task does not possess an associated TaskConfig, this entry will be simply skipped.
-# It is populated automatically.
+# It is populated automatically, do not modify.
 _task_class_registry = {}
 
 # Automatic population of registries