Fix tests

tests/assets/configs/dmz_network.yaml

@@ -65,7 +65,7 @@ game:
 agents:
   - ref: client_1_green_user
     team: GREEN
-    type: GreenWebBrowsingAgent
+    type: probabilistic_agent
     observation_space:
       type: UC2GreenObservation
     action_space:

tests/e2e_integration_tests/test_primaite_session.py

@@ -21,15 +21,15 @@ class TestPrimaiteSession:
                 raise AssertionError
 
             assert session is not None
-            assert session.game.simulation
-            assert len(session.game.agents) == 3
-            assert len(session.game.rl_agents) == 1
+            assert session.env.game.simulation
+            assert len(session.env.game.agents) == 3
+            assert len(session.env.game.rl_agents) == 1
 
             assert session.policy
             assert session.env
 
-            assert session.game.simulation.network
-            assert len(session.game.simulation.network.nodes) == 10
+            assert session.env.game.simulation.network
+            assert len(session.env.game.simulation.network.nodes) == 10
 
     @pytest.mark.parametrize("temp_primaite_session", [[CFG_PATH]], indirect=True)
     def test_start_session(self, temp_primaite_session):