Merge pull request #192 from edbeeching/fix-numpy

Set numpy version, bump release
edbeeching · Jun 25, 2024 · d1ecffe · d1ecffe
2 parents 89a0b5a + 2bd6139
commit d1ecffe
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/godot_rl/wrappers/ray_wrapper.py b/godot_rl/wrappers/ray_wrapper.py
@@ -127,7 +127,7 @@ def rllib_training(args, extras):
             checkpoint_freq=checkpoint_freq,
             checkpoint_at_end=not args.eval,
             restore=args.restore,
-            local_dir=os.path.abspath(args.experiment_dir) or os.path.abspath("logs/rllib"),
+            storage_path=os.path.abspath(args.experiment_dir) or os.path.abspath("logs/rllib"),
             trial_name_creator=lambda trial: (
                 f"{args.experiment_name}" if args.experiment_name else f"{trial.trainable_name}_{trial.trial_id}"
             ),

diff --git a/pyproject.toml b/pyproject.toml
@@ -3,7 +3,7 @@ requires = ["setuptools>=61.0"]
 build-backend = "setuptools.build_meta"
 [project]
 name = "godot_rl"
-version = "0.8.0"
+version = "0.8.1"
 authors = [
   { name="Edward Beeching", email="edbeeching@gmail.com" },
 ]

diff --git a/setup.cfg b/setup.cfg
@@ -11,7 +11,7 @@ classifiers =
 [options]
 packages = find:
 install_requires =
-    numpy
+    numpy<2
     tensorboard
     wget
     huggingface_hub>=0.10