test

logicalclocks · Jul 3, 2024 · 26a1e06 · 26a1e06
1 parent 87d70e3
commit 26a1e06
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/python/tests/test_model.py b/python/tests/test_model.py
@@ -202,7 +202,7 @@ def test_deploy(self, mocker, backend_fixtures):
             inference_logger=inference_logger,
             inference_batcher=inference_batcher,
             transformer=transformer,
-            api_protocol=p_json["api_protocol"],
+            environment=p_json["environment_dto"]["name"],
         )
 
         # Assert
@@ -218,6 +218,7 @@ def test_deploy(self, mocker, backend_fixtures):
             inference_batcher=inference_batcher,
             transformer=transformer,
             api_protocol=p_json["api_protocol"],
+            environment=p_json["environment_dto"]["name"],
         )
         mock_predictor.deploy.assert_called_once()