Fixed some more failing tests. Added six as an explicit dependency due to tensorboard requirements.

Author: Miguel Alonso Jr

ml-agents/mlagents/trainers/tests/torch_entities/test_hybrid.py

@@ -90,7 +90,7 @@ def test_hybrid_sac(action_size):
         buffer_init_steps=0,
     )
     config = attr.evolve(
-        SAC_TORCH_CONFIG, hyperparameters=new_hyperparams, max_steps=8000
+        SAC_TORCH_CONFIG, hyperparameters=new_hyperparams, max_steps=10000
     )
     check_environment_trains(env, {BRAIN_NAME: config}, success_threshold=0.9)
 

ml-agents/setup.py

@@ -70,7 +70,9 @@ def run(self):
         # https://github.com/pytorch/pytorch/issues/50014
         "torch>=1.8.0,<=1.11.0;(platform_system!='Windows' and python_version>='3.9')",
         "torch>=1.6.0,<1.9.0;(platform_system!='Windows' and python_version<'3.9')",
-        "tensorboard>=1.15",
+        "tensorboard>=2.14",
+        # adding six explicit dependency since tensorboard needs it but doesn't declare it as a dep
+        "six>=1.16",
         # cattrs 1.1.0 dropped support for python 3.6, but 1.0.0 doesn't work for python 3.9
         # Since there's no version that supports both, we have to draw the line somwehere.
         "cattrs<1.1.0; python_version<'3.8'",