e2e tests: Fix RayJobClient package

opendatahub-io · Oct 23, 2024 · e206b1d · e206b1d
1 parent 81b4bfe
commit e206b1d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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"
    ]
   },