From cf4e1369794c1b628f1891f46658382f30162f7c Mon Sep 17 00:00:00 2001
From: Sam Shi <yixinshi@gmail.com>
Date: Fri, 17 Aug 2018 16:23:38 -0500
Subject: [PATCH] fix a typo and remove a too restrictive limit (#1382)

---
 kubeflow/examples/prototypes/tf-batch-predict.jsonnet | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/kubeflow/examples/prototypes/tf-batch-predict.jsonnet b/kubeflow/examples/prototypes/tf-batch-predict.jsonnet
index 6a85f0245b2..876e6d1216f 100644
--- a/kubeflow/examples/prototypes/tf-batch-predict.jsonnet
+++ b/kubeflow/examples/prototypes/tf-batch-predict.jsonnet
@@ -87,9 +87,7 @@ local tfBatchPredictBase = {
 
                 resources: {
                   limits: {
-                    memory: "4Gi",
-                    cpu: "4",
-                    [if base.params.numGpus > 0 then "nvidia/com/gpu"]: base.params.numGpus,
+                    [if base.params.numGpus > 0 then "nvidia.com/gpu"]: base.params.numGpus,
                   },
                 },