Fix PGMORL example (#104)

* Fix PGMORL example * Numpy 2 breaks pymoo
LucasAlegre · Jul 16, 2024 · f3d7c95 · f3d7c95 · twobox · Aug 8, 2024
1 parent 349ecd5
commit f3d7c95
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/examples/pgmorl_halfcheetah.py b/examples/pgmorl_halfcheetah.py
@@ -18,6 +18,7 @@
     )
     algo.train(
         total_timesteps=int(5e6),
+        eval_env=make_env(env_id, 42, 0, "PGMORL_eval_env", gamma=0.995)(),
         ref_point=np.array([0.0, -5.0]),
         known_pareto_front=None,
     )

diff --git a/pyproject.toml b/pyproject.toml
@@ -23,7 +23,7 @@ classifiers = [
 dependencies = [
     "mo-gymnasium >=1.0.1",
     "gymnasium>=0.28.1,<0.30",
-    "numpy >=1.21.0",
+    "numpy >=1.21.0,<2.0.0",
     "torch >=1.12.0",
     "pygame >=2.1.0",
     "scipy >=1.7.3",