fix badly hardcoded integration truths

zapata-engineering · Sep 27, 2024 · 5cf1360 · 5cf1360
1 parent e159fb1
commit 5cf1360
Showing 1 changed file with 3 additions and 6 deletions.
diff --git 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,
                     },