Remove upper bound limit

Author: maryamziaa

.github/workflows/colab.yml

@@ -16,7 +16,7 @@ on:
 
 jobs:
   colab:
-    runs-on: ubuntu-latest
+    runs-on: ubuntu-22.04
     env:
         COLAB_ALWAYS_INSTALL_XVFB: 1
         QLEARNING_NUM_TRAINING_STEPS: 5

colab/Colab_UnityEnvironment_1_Run.ipynb

@@ -217,7 +217,7 @@
     "\n",
     "from mlagents_envs.registry import default_registry\n",
     "\n",
-    "env = default_registry[env_id].make(timeout_wait=120)"
+    "env = default_registry[env_id].make()"
    ],
    "execution_count": 3,
    "outputs": [

ml-agents/setup.py

@@ -62,7 +62,7 @@ def run(self):
         "Pillow>=4.2.1",
         "protobuf>=3.6,<3.21",
         "pyyaml>=3.1.0",
-        "torch>=2.1.1,<=2.8.0",
+        "torch>=2.1.1",
         "tensorboard>=2.14",
         # adding six explicit dependency since tensorboard needs it but doesn't declare it as a dep
         "six>=1.16",