diff --git a/.jenkins/validate_tutorials_built.py b/.jenkins/validate_tutorials_built.py index ef4b41a6bcc..bf42e0845ff 100644 --- a/.jenkins/validate_tutorials_built.py +++ b/.jenkins/validate_tutorials_built.py @@ -24,7 +24,6 @@ "examples_autograd/polynomial_custom_function", "forward_ad_usage", "parametrizations", - "reinforcement_q_learning", "mnist_train_nas", # used by ax_multiobjective_nas_tutorial.py "fx_conv_bn_fuser", "super_resolution_with_onnxruntime", diff --git a/intermediate_source/reinforcement_q_learning.py b/intermediate_source/reinforcement_q_learning.py index cb9abc229c9..6a03d61f79e 100644 --- a/intermediate_source/reinforcement_q_learning.py +++ b/intermediate_source/reinforcement_q_learning.py @@ -46,7 +46,12 @@ First, let's import needed packages. Firstly, we need `gym `__ for the environment -(Install using `pip install gym`). + +.. code-block:: bash + + %%bash + pip3 install gym[classic_control] + We'll also use the following from PyTorch: - neural networks (``torch.nn``) diff --git a/requirements.txt b/requirements.txt index 1e70f8c4268..74e2da5fad3 100644 --- a/requirements.txt +++ b/requirements.txt @@ -43,6 +43,7 @@ scikit-image scipy pillow==9.0.1 wget -gym==0.24.0 -gym-super-mario-bros==7.3.0 +gym==0.25.1 +gym-super-mario-bros==7.4.0 timm +pygame==2.1.2