Fix ray tests after rebase

Signed-off-by: Fabio Graetz <[email protected]>
flyteorg · May 24, 2024 · e4bada1 · e4bada1
1 parent 082178d
commit e4bada1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/flyteplugins/go/tasks/plugins/k8s/ray/ray_test.go b/flyteplugins/go/tasks/plugins/k8s/ray/ray_test.go
@@ -776,8 +776,8 @@ func TestGetTaskPhaseIncreasePhaseVersion(t *testing.T) {
 	}
 	pluginCtx := newPluginContext(pluginState)
 
-	rayObject := &rayv1alpha1.RayJob{}
-	rayObject.Status.JobDeploymentStatus = rayv1alpha1.JobDeploymentStatusInitializing
+	rayObject := &rayv1.RayJob{}
+	rayObject.Status.JobDeploymentStatus = rayv1.JobDeploymentStatusInitializing
 	phaseInfo, err := rayJobResourceHandler.GetTaskPhase(ctx, pluginCtx, rayObject)
 
 	assert.NoError(t, err)