From f3ebfbd5769a9d56bc167e501d077c79f80ace8b Mon Sep 17 00:00:00 2001
From: Matteo Bettini <mb2389@cl.cam.ac.uk>
Date: Sat, 21 Oct 2023 17:44:47 +0100
Subject: [PATCH] amend

---
 benchmarl/__init__.py | 10 ----------
 docs/source/api.rst   |  5 ++++-
 2 files changed, 4 insertions(+), 11 deletions(-)

diff --git a/benchmarl/__init__.py b/benchmarl/__init__.py
index 5fad7aab..a7bc3127 100644
--- a/benchmarl/__init__.py
+++ b/benchmarl/__init__.py
@@ -6,16 +6,6 @@
 
 import importlib
 
-from benchmarl import (
-    algorithms,
-    benchmark,
-    environments,
-    eval_results,
-    experiment,
-    models,
-    utils,
-)
-
 
 _has_hydra = importlib.util.find_spec("hydra") is not None
 
diff --git a/docs/source/api.rst b/docs/source/api.rst
index 4727592c..55aec41c 100644
--- a/docs/source/api.rst
+++ b/docs/source/api.rst
@@ -4,4 +4,7 @@ API
 .. autosummary::
    :toctree: generated
 
-   benchmarl
+   benchmarl.algorithms
+   benchmarl.models
+   benchmarl.experiment
+   benchmarl.environments