Skip to content

Commit 8a6bf36

Browse files
authored
Re-enable reinforcement_q_learning tutorial (#2067)
By adding pygame to the requirements and bumping gym to 0.25.0 (to account for changes introduced by #2026 ) Add cell for installing `gym` on Colab
1 parent 1c11ecf commit 8a6bf36

File tree

3 files changed

+9
-4
lines changed

3 files changed

+9
-4
lines changed

.jenkins/validate_tutorials_built.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,6 @@
2424
"examples_autograd/polynomial_custom_function",
2525
"forward_ad_usage",
2626
"parametrizations",
27-
"reinforcement_q_learning",
2827
"mnist_train_nas", # used by ax_multiobjective_nas_tutorial.py
2928
"fx_conv_bn_fuser",
3029
"super_resolution_with_onnxruntime",

intermediate_source/reinforcement_q_learning.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -46,7 +46,12 @@
4646
4747
First, let's import needed packages. Firstly, we need
4848
`gym <https://github.com/openai/gym>`__ for the environment
49-
(Install using `pip install gym`).
49+
50+
.. code-block:: bash
51+
52+
%%bash
53+
pip3 install gym[classic_control]
54+
5055
We'll also use the following from PyTorch:
5156
5257
- neural networks (``torch.nn``)

requirements.txt

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -43,6 +43,7 @@ scikit-image
4343
scipy
4444
pillow==9.0.1
4545
wget
46-
gym==0.24.0
47-
gym-super-mario-bros==7.3.0
46+
gym==0.25.1
47+
gym-super-mario-bros==7.4.0
4848
timm
49+
pygame==2.1.2

0 commit comments

Comments
 (0)