amend

facebookresearch · Nov 2, 2023 · 19af36a · 19af36a
1 parent 4896984
commit 19af36a
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 5 deletions.
diff --git a/benchmarl/__init__.py b/benchmarl/__init__.py
@@ -6,6 +6,14 @@
 
 import importlib
 
+import benchmarl.algorithms
+import benchmarl.benchmark
+import benchmarl.environments
+import benchmarl.eval_results
+import benchmarl.experiment
+import benchmarl.models
+import benchmarl.utils
+
 
 _has_hydra = importlib.util.find_spec("hydra") is not None
 

diff --git a/benchmarl/hydra_config.py b/benchmarl/hydra_config.py
@@ -3,8 +3,12 @@
 #  This source code is licensed under the license found in the
 #  LICENSE file in the root directory of this source tree.
 #
+import importlib
 
-from omegaconf import DictConfig, OmegaConf
+_has_hydra = importlib.util.find_spec("hydra") is not None
+
+if _has_hydra:
+    from omegaconf import DictConfig, OmegaConf
 
 from benchmarl.algorithms.common import AlgorithmConfig
 from benchmarl.environments import Task, task_config_registry

diff --git a/docs/source/api.rst b/docs/source/api.rst
@@ -4,7 +4,4 @@ API
 .. autosummary::
    :toctree: generated
 
-   benchmarl.algorithms
-   benchmarl.models
-   benchmarl.experiment
-   benchmarl.environments
+   benchmarl