From f6f26f14e823fa51e22a2c9d3716894a52dae336 Mon Sep 17 00:00:00 2001
From: Daniela Plascencia <daniela.plascencia@linux.intel.com>
Date: Fri, 25 Jan 2019 15:24:09 -0600
Subject: [PATCH] prototypes: sets up ks prototypes to receive image names as
 parameter

The image name was hardcoded. We should receive a variable with the
image name instead, this will let the end user select which DLRS
flavour (mkl or oss) to use.

Signed-off-by: Daniela Plascencia <daniela.plascencia@linux.intel.com>
---
 .../dlrs-tfjob/dlrs-bench/prototypes/dlrs-alexnet.jsonnet    | 5 +++--
 .../dlrs-tfjob/dlrs-bench/prototypes/dlrs-resnet50.jsonnet   | 5 +++--
 2 files changed, 6 insertions(+), 4 deletions(-)

diff --git a/stacks/dlrs/kubeflow/dlrs-tfjob/dlrs-bench/prototypes/dlrs-alexnet.jsonnet b/stacks/dlrs/kubeflow/dlrs-tfjob/dlrs-bench/prototypes/dlrs-alexnet.jsonnet
index 6501212..446de49 100644
--- a/stacks/dlrs/kubeflow/dlrs-tfjob/dlrs-bench/prototypes/dlrs-alexnet.jsonnet
+++ b/stacks/dlrs/kubeflow/dlrs-tfjob/dlrs-bench/prototypes/dlrs-alexnet.jsonnet
@@ -3,16 +3,17 @@
 // @description dlrs multi-node benchmark
 // @shortDescription A multi-node DLaaS benchmark
 // @param name string Name for the job.
+// @param image string DLRS Image name.
 
 local k = import "k.libsonnet";
 
 local name = params.name;
 local namespace = env.namespace;
-local image = "hub.docker.com/clearlinux/stacks-dlrs-kubeflow:latest";
+local image = params.image;
 local replicas = 3;
 
 local tfjob = {
-  apiVersion: "kubeflow.org/v1alpha2",
+  apiVersion: "kubeflow.org/v1beta1",
   kind: "TFJob",
   metadata: {
     name: name,
diff --git a/stacks/dlrs/kubeflow/dlrs-tfjob/dlrs-bench/prototypes/dlrs-resnet50.jsonnet b/stacks/dlrs/kubeflow/dlrs-tfjob/dlrs-bench/prototypes/dlrs-resnet50.jsonnet
index ef4c7e4..42bb23f 100644
--- a/stacks/dlrs/kubeflow/dlrs-tfjob/dlrs-bench/prototypes/dlrs-resnet50.jsonnet
+++ b/stacks/dlrs/kubeflow/dlrs-tfjob/dlrs-bench/prototypes/dlrs-resnet50.jsonnet
@@ -3,16 +3,17 @@
 // @description dlrs multi-node benchmark
 // @shortDescription A multi-node DLaaS benchmark
 // @param name string Name for the job.
+// @param image string DLRS Image name.
 
 local k = import "k.libsonnet";
 
 local name = params.name;
 local namespace = env.namespace;
-local image = "hub.docker.com/clearlinux/stacks-dlrs-kubeflow:latest";
+local image = params.image;
 local replicas = 3;
 
 local tfjob = {
-  apiVersion: "kubeflow.org/v1alpha2",
+  apiVersion: "kubeflow.org/v1beta1",
   kind: "TFJob",
   metadata: {
     name: name,