From 5cf1360e636c9cdf3fa8fbf35e27346244fdac65 Mon Sep 17 00:00:00 2001 From: Sebastian Morawiec Date: Fri, 27 Sep 2024 12:37:00 +0200 Subject: [PATCH] fix badly hardcoded integration truths --- .../tests/integration/ray/test_integration.py | 9 +++------ 1 file changed, 3 insertions(+), 6 deletions(-) diff --git a/projects/orquestra-sdk/tests/integration/ray/test_integration.py b/projects/orquestra-sdk/tests/integration/ray/test_integration.py index fd86e0654..08be65068 100644 --- a/projects/orquestra-sdk/tests/integration/ray/test_integration.py +++ b/projects/orquestra-sdk/tests/integration/ray/test_integration.py @@ -28,6 +28,7 @@ from orquestra import sdk from orquestra.sdk._client._base._config._settings import LOCAL_RUNTIME_CONFIGURATION +from orquestra.sdk._client._base._docker_images import DEFAULT_WORKER_IMAGE from orquestra.sdk._client._base._testing import _example_wfs, _ipc from orquestra.sdk._client._base._testing._example_wfs import ( workflow_parametrised_with_resources, @@ -975,17 +976,13 @@ def test_setting_resources( ( None, None, - { - "image:hub.stage.nexus.orquestra.io/zapatacomputing/orquestra-sdk-base:worker-1.0.0a1": 1 # noqa: E501 - }, + {f"image:{DEFAULT_WORKER_IMAGE}": 1}, {}, ), ( None, 1, - { - "image:hub.stage.nexus.orquestra.io/zapatacomputing/orquestra-sdk-base:worker-1.0.0a1-cuda": 1 # noqa: E501 - }, + {f"image:{DEFAULT_WORKER_IMAGE}-cuda": 1}, { "num_gpus": 1, },