Skip to content

Commit d32f067

Browse files
committed
fix python3.11 test
1 parent 4f110d6 commit d32f067

1 file changed

Lines changed: 6 additions & 1 deletion

File tree

setup.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -65,14 +65,19 @@ def get_extra_requires() -> dict:
6565
"evaluate",
6666
"icetk",
6767
],
68+
"nlp_test": [
69+
"transformers",
70+
"datasets",
71+
"evaluate",
72+
],
6873
"selfplay": ["ray[default]", "ray[serve]", "pettingzoo[classic]", "trueskill"],
6974
"retro": ["gym-retro"],
7075
"super_mario": ["gym-super-mario-bros"],
7176
"atari": ["gymnasium[atari]", "gymnasium[accept-rom-license]"],
7277
}
7378
req["test"].extend(req["selfplay"])
7479
req["test"].extend(req["atari"])
75-
req["test"].extend(req["nlp"])
80+
req["test"].extend(req["nlp_test"])
7681
return req
7782

7883

0 commit comments

Comments
 (0)