Skip to content

Commit 7a67cde

Browse files
authored
Fix ray requirements for tutorial (Farama-Foundation#1139)
1 parent 629465f commit 7a67cde

File tree

4 files changed

+4
-4
lines changed

4 files changed

+4
-4
lines changed

.github/workflows/linux-tutorials-test.yml

+1-1
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@ jobs:
1717
fail-fast: false
1818
matrix:
1919
python-version: ['3.8', '3.9', '3.10', '3.11']
20-
tutorial: [Tianshou, CustomEnvironment, CleanRL, SB3/kaz, SB3/waterworld, SB3/connect_four, SB3/test, AgileRL] # TODO: add back Ray once next release after 2.6.2
20+
tutorial: [Tianshou, CustomEnvironment, CleanRL, SB3/kaz, SB3/waterworld, SB3/connect_four, SB3/test, AgileRL, Ray]
2121
steps:
2222
- uses: actions/checkout@v4
2323
- name: Set up Python ${{ matrix.python-version }}

tutorials/Ray/requirements.txt

+1-1
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
PettingZoo[classic,butterfly]>=1.24.0
22
Pillow>=9.4.0
3-
ray[rllib]>=2.7.0
3+
ray[rllib]==2.7.0
44
SuperSuit>=3.9.0
55
torch>=1.13.1
66
tensorflow-probability>=0.19.0

tutorials/Ray/rllib_leduc_holdem.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -123,7 +123,7 @@ def env_creator():
123123
tune.run(
124124
alg_name,
125125
name="DQN",
126-
stop={"timesteps_total": 10000000},
126+
stop={"timesteps_total": 10000000 if not os.environ.get("CI") else 50000},
127127
checkpoint_freq=10,
128128
config=config.to_dict(),
129129
)

tutorials/Ray/rllib_pistonball.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -98,7 +98,7 @@ def env_creator(args):
9898
tune.run(
9999
"PPO",
100100
name="PPO",
101-
stop={"timesteps_total": 5000000},
101+
stop={"timesteps_total": 5000000 if not os.environ.get("CI") else 50000},
102102
checkpoint_freq=10,
103103
local_dir="~/ray_results/" + env_name,
104104
config=config.to_dict(),

0 commit comments

Comments
 (0)