diff --git a/pathmind/simulation.py b/pathmind/simulation.py index 3bc6d7b..697a414 100644 --- a/pathmind/simulation.py +++ b/pathmind/simulation.py @@ -196,7 +196,6 @@ def train( """ env_name = str(self.__class__).split("'")[1] - multi_agent = self.number_of_agents() > 1 if not observation_yaml: write_observation_yaml(self, base_folder) @@ -216,10 +215,8 @@ def train( cmd = f"""curl -i -XPOST \ -H "X-PM-API-TOKEN: {token}" \ -F 'file=@training.zip' \ - -F 'isPathmindSimulation=true' \ -F 'env={env_name}' \ -F 'start=true' \ - -F 'multiAgent={multi_agent}' \ -F 'obsSelection={obs_yaml}' \ https://api.pathmind.com/py/upload """ diff --git a/tests/test_train_simulation.py b/tests/test_train_simulation.py index 70ca98f..a2b49d4 100644 --- a/tests/test_train_simulation.py +++ b/tests/test_train_simulation.py @@ -12,8 +12,8 @@ def test_training(): simulation.train() +@pytest.mark.xfail(reason="Multi-agent hasn't been released to prod yet") def test_multi_training(): - pytest.skip("Needs multi-agent training to work on web app") simulation = MultiMouseAndCheese() simulation.train()