From 2bd6139eabe5c06152ce13de835b5409591ddb4c Mon Sep 17 00:00:00 2001
From: Ivan-267 <61947090+Ivan-267@users.noreply.github.com>
Date: Tue, 25 Jun 2024 17:45:50 +0200
Subject: [PATCH] Update ray_wrapper.py

Updates local_dir to storage_path
---
 godot_rl/wrappers/ray_wrapper.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/godot_rl/wrappers/ray_wrapper.py b/godot_rl/wrappers/ray_wrapper.py
index ab86cd1a..0562c57e 100644
--- 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}"
             ),