Skip to content

Commit d474e6b

Browse files
committed
test: update
Signed-off-by: Sugam Devare <sdevare@nvidia.com>
1 parent b1e23b4 commit d474e6b

1 file changed

Lines changed: 2 additions & 4 deletions

File tree

tests/unit_tests/test_server_utils.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -194,9 +194,7 @@ def test_initialize_ray_with_address(self, monkeypatch: MonkeyPatch) -> None:
194194

195195
ray_is_initialized_mock.assert_called_once()
196196
get_global_config_dict_mock.assert_called_once()
197-
ray_init_mock.assert_called_once_with(
198-
address="ray://test-address:10001", ignore_reinit_error=True, runtime_env={"working_dir": PARENT_DIR}
199-
)
197+
ray_init_mock.assert_called_once_with(address="ray://test-address:10001", ignore_reinit_error=True)
200198

201199
def test_initialize_ray_without_address(self, monkeypatch: MonkeyPatch) -> None:
202200
ray_is_initialized_mock = self._mock_ray_return_value(monkeypatch, False)
@@ -214,4 +212,4 @@ def test_initialize_ray_without_address(self, monkeypatch: MonkeyPatch) -> None:
214212
ray_is_initialized_mock.assert_called_once()
215213
get_global_config_dict_mock.assert_called_once()
216214

217-
ray_init_mock.assert_called_once_with(ignore_reinit_error=True, runtime_env={"working_dir": PARENT_DIR})
215+
ray_init_mock.assert_called_once_with(ignore_reinit_error=True)

0 commit comments

Comments
 (0)