e2e tests: Fix RayJobClient package

sutaakar · openshift-merge-bot[bot] · commit e206b1d885c3 · 2024-10-23T07:56:57.000Z
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"
    ]
   },

Original file line number	Diff line number	Diff line change
`@@ -14,7 +14,7 @@`
`14`	`14`	`"\n",`
`15`	`15`	`"# Import pieces from codeflare-sdk\n",`
`16`	`16`	`"from codeflare_sdk import Cluster, ClusterConfiguration, TokenAuthentication\n",`
`17`		`- "from codeflare_sdk.job import RayJobClient\n",`
	`17`	`+ "from codeflare_sdk.ray.client import RayJobClient\n",`
`18`	`18`	`"from time import sleep"`
`19`	`19`	`]`
`20`	`20`	`},`