fix wrong api version when delete pytorchjob (#179)

kubeflow · Jun 19, 2019 · 213df9c · 213df9c
1 parent e775742
commit 213df9c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pkg/controller.v1/pytorch/job.go b/pkg/controller.v1/pytorch/job.go
@@ -197,7 +197,7 @@ func (pc *PyTorchController) cleanupPyTorchJob(job *pyv1.PyTorchJob) error {
 
 // deletePyTorchJob deletes the given PyTorchJob.
 func (pc *PyTorchController) deletePyTorchJob(job *pyv1.PyTorchJob) error {
-	return pc.jobClientSet.KubeflowV1beta2().PyTorchJobs(job.Namespace).Delete(job.Name, &metav1.DeleteOptions{})
+	return pc.jobClientSet.KubeflowV1().PyTorchJobs(job.Namespace).Delete(job.Name, &metav1.DeleteOptions{})
 }
 
 func getTotalReplicas(job *pyv1.PyTorchJob) int32 {