Attempt to fix module not found issues

morl_baselines/__init__.py

@@ -1,4 +1,6 @@
 """MORL-Baselines contains various MORL algorithms and utility functions."""
-
+import morl_baselines.common
+import morl_baselines.multi_policy
+import morl_baselines.single_policy
 
 __version__ = "1.0.0"

morl_baselines/multi_policy/__init__.py

@@ -1 +1,10 @@
 """Multi-policy algorithms."""
+import morl_baselines.multi_policy.capql
+import morl_baselines.multi_policy.envelope
+import morl_baselines.multi_policy.gpi_pd
+import morl_baselines.multi_policy.linear_support
+import morl_baselines.multi_policy.morld
+import morl_baselines.multi_policy.multi_policy_moqlearning
+import morl_baselines.multi_policy.pareto_q_learning
+import morl_baselines.multi_policy.pcn
+import morl_baselines.multi_policy.pgmorl

morl_baselines/multi_policy/capql/__init__.py

@@ -1 +1,2 @@
 """Concave-Augmented Pareto Q-Learning (CAPQL) implementation."""
+import morl_baselines.multi_policy.capql.capql

morl_baselines/multi_policy/envelope/__init__.py

@@ -1 +1,2 @@
 """Envelope Q-Learning (EQL) implementation."""
+import morl_baselines.multi_policy.envelope.envelope

morl_baselines/multi_policy/gpi_pd/__init__.py

@@ -1 +1,3 @@
 """GPI-Prioritized Dyna."""
+import morl_baselines.multi_policy.gpi_pd.gpi_pd
+import morl_baselines.multi_policy.gpi_pd.gpi_pd_continuous_action

morl_baselines/multi_policy/linear_support/__init__.py

@@ -1 +1,2 @@
 """Linear support algorithm."""
+import morl_baselines.multi_policy.linear_support.linear_support

morl_baselines/multi_policy/morld/__init__.py

@@ -2,3 +2,4 @@
 
 See Felten, Talbi & Danoy (2024): https://arxiv.org/abs/2311.12495.
 """
+import morl_baselines.multi_policy.morld.morld

morl_baselines/multi_policy/multi_policy_moqlearning/__init__.py

@@ -1 +1,2 @@
 """Outer-loop MOQ-learning algorithm."""
+import morl_baselines.multi_policy.multi_policy_moqlearning.mp_mo_q_learning

morl_baselines/multi_policy/pareto_q_learning/__init__.py

@@ -1 +1,2 @@
 """Pareto Q-Learning."""
+import morl_baselines.multi_policy.pareto_q_learning.pql

morl_baselines/multi_policy/pcn/__init__.py

@@ -1 +1,2 @@
 """Pareto Conditioned Networks (PCN) for multi-policy learning."""
+import morl_baselines.multi_policy.pcn.pcn

morl_baselines/multi_policy/pgmorl/__init__.py

@@ -1 +1,2 @@
 """Prediction Guided Multi-Objective Reinforcement Learning (PGMORL)."""
+import morl_baselines.multi_policy.pgmorl.pgmorl