Skip to content

Commit e0b3b59

Browse files
Update setup.py and test
1 parent 09a512b commit e0b3b59

File tree

2 files changed

+9
-6
lines changed

2 files changed

+9
-6
lines changed

setup.py

+8-5
Original file line numberDiff line numberDiff line change
@@ -12,14 +12,17 @@
1212
package_data = {"": ["*"]}
1313

1414
extras_require = [
15-
"ray[default,rllib]==1.3.0",
16-
"tensorflow==2.2.0",
15+
"ray[default,rllib]==1.0.0",
16+
"tensorflow==2.5.0",
1717
"pillow>=6.1.0",
1818
"requests==2.22.0",
1919
"configspace==0.4.10",
20-
"scipy>=1.3.0",
21-
"pandas==0.25.0",
22-
"gym[atari]==0.18",
20+
"scipy>=1.7.0",
21+
"pandas>=0.25.0",
22+
"gym[atari]==0.20",
23+
"matplotlib",
24+
"opencv-python==4.2.0.34", # due to https://stackoverflow.com/questions/72706073/attributeerror-partially-initialized-module-cv2-has-no-attribute-gapi-wip-gs
25+
"opencv-python-headless==4.2.0.34", # and https://stackoverflow.com/questions/70537488/cannot-import-name-registermattype-from-cv2-cv2
2326
]
2427

2528
extras_require_disc = [

tests/test_run_experiments.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -53,7 +53,7 @@ def test_dqn_test_expt(self):
5353
err_msg="Expected training timesteps, episode_reward, episode_len after 10,000 timesteps to be within 10% of [10000., 80.0, 80.0].",
5454
)
5555

56-
exit_code = os.system("rm dqn_test_expt_0*.csv")
56+
exit_code = os.system("rm -f dqn_test_expt_0*.csv")
5757
assert exit_code == 0
5858

5959
# Similar thing is tested above. These tests are time consuming, so rather have only a few of them.

0 commit comments

Comments
 (0)