From e206b1d885c3b010f6f2ad66569321093a20d33e Mon Sep 17 00:00:00 2001 From: Karel Suta Date: Tue, 22 Oct 2024 15:18:57 +0200 Subject: [PATCH] e2e tests: Fix RayJobClient package --- tests/odh/resources/mnist_hpo_raytune.ipynb | 2 +- tests/odh/resources/mnist_ray_mini.ipynb | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/odh/resources/mnist_hpo_raytune.ipynb b/tests/odh/resources/mnist_hpo_raytune.ipynb index 6179d213..4d8c2343 100644 --- a/tests/odh/resources/mnist_hpo_raytune.ipynb +++ b/tests/odh/resources/mnist_hpo_raytune.ipynb @@ -14,7 +14,7 @@ "\n", "# Import pieces from codeflare-sdk\n", "from codeflare_sdk import Cluster, ClusterConfiguration, TokenAuthentication\n", - "from codeflare_sdk.job import RayJobClient\n", + "from codeflare_sdk.ray.client import RayJobClient\n", "from time import sleep" ] }, diff --git a/tests/odh/resources/mnist_ray_mini.ipynb b/tests/odh/resources/mnist_ray_mini.ipynb index 2e7e3ba7..010ce510 100644 --- a/tests/odh/resources/mnist_ray_mini.ipynb +++ b/tests/odh/resources/mnist_ray_mini.ipynb @@ -14,7 +14,7 @@ "\n", "# Import pieces from codeflare-sdk\n", "from codeflare_sdk import Cluster, ClusterConfiguration, TokenAuthentication\n", - "from codeflare_sdk.job import RayJobClient\n", + "from codeflare_sdk.ray.client import RayJobClient\n", "from time import sleep" ] },