Cleaning up and updating requirements

environment.yml

@@ -70,7 +70,7 @@ dependencies:
     - crowdai-api==0.1.21
     - cycler==0.10.0
     - filelock==3.0.12
-    - flatland-rl==2.2.1
+    - flatland-rl==2.2.2
     - future==0.17.1
     - gym==0.14.0
     - idna==2.8
@@ -102,10 +102,11 @@ dependencies:
     - timeout-decorator==0.4.1
     - toml==0.10.0
     - torch==1.5.0
-    - tox==3.13.2
     - urllib3==1.25.3
     - ushlex==0.99.1
-    - virtualenv==16.7.2
     - wcwidth==0.1.7
     - xarray==0.12.3
     - zipp==0.5.2
+    - psutil==5.7.2
+    - tensorboard==2.3.0
+    - tensorboardX==2.1