Skip to content

Commit

Permalink
Update ray_wrapper.py
Browse files Browse the repository at this point in the history
Updates local_dir to storage_path
  • Loading branch information
Ivan-267 authored Jun 25, 2024
1 parent 9b6a747 commit 2bd6139
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion godot_rl/wrappers/ray_wrapper.py
Original file line number Diff line number Diff line change
Expand Up @@ -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}"
),
Expand Down

0 comments on commit 2bd6139

Please sign in to comment.