From 9657d56a2a0f625da2ced60356495b031fc5e246 Mon Sep 17 00:00:00 2001
From: Johnu George <johnugeorge109@gmail.com>
Date: Thu, 8 Nov 2018 13:50:10 +0530
Subject: [PATCH] Adding validation for v1beta1 apis

---
 pkg/apis/pytorch/validation/validation.go | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pkg/apis/pytorch/validation/validation.go b/pkg/apis/pytorch/validation/validation.go
index 0abaab921..833461af3 100644
--- a/pkg/apis/pytorch/validation/validation.go
+++ b/pkg/apis/pytorch/validation/validation.go
@@ -170,7 +170,7 @@ func ValidateBetaOnePyTorchJobSpec(c *torchv1beta1.PyTorchJobSpec) error {
 				log.Warn("Image is undefined in the container")
 				return fmt.Errorf("PyTorchJobSpec is not valid")
 			}
-			if container.Name == torchv2.DefaultContainerName {
+			if container.Name == torchv1beta1.DefaultContainerName {
 				defaultContainerPresent = true
 			}
 		}