fix(components): Update watson ml sdk with the Aug 2020 release (#4331)

kubeflow · Aug 7, 2020 · 25ee59d · 25ee59d
1 parent d135693
commit 25ee59d
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 5 deletions.
diff --git a/components/ibm-components/watson/deploy/Dockerfile b/components/ibm-components/watson/deploy/Dockerfile
@@ -5,7 +5,7 @@ RUN mkdir /app
 RUN mkdir /app/secrets
 
 # Watson studio and machine learning python client
-RUN pip install watson-machine-learning-client-V4 minio
+RUN pip install watson-machine-learning-client-V4>=1.0.110 minio
 
 # Python functions with endpoints to Watson Machine Learning
 COPY src/wml-deploy.py /app
diff --git a/components/ibm-components/watson/store/Dockerfile b/components/ibm-components/watson/store/Dockerfile
@@ -5,7 +5,7 @@ RUN mkdir /app
 RUN mkdir /app/secrets
 
 # Watson studio and machine learning python client
-RUN pip install watson-machine-learning-client-V4 minio
+RUN pip install watson-machine-learning-client-V4>=1.0.110 minio
 
 # Python functions with endpoints to Watson Machine Learning
 COPY src/wml-store.py /app
diff --git a/components/ibm-components/watson/train/Dockerfile b/components/ibm-components/watson/train/Dockerfile
@@ -5,7 +5,7 @@ RUN mkdir /app
 RUN mkdir /app/secrets
 
 # Watson studio and machine learning python client
-RUN pip install watson-machine-learning-client-V4 minio
+RUN pip install watson-machine-learning-client-V4>=1.0.110 minio
 
 # Python functions with endpoints to Watson Machine Learning
 COPY src/wml-train.py /app
diff --git a/components/ibm-components/watson/train/src/wml-train.py b/components/ibm-components/watson/train/src/wml-train.py
@@ -191,9 +191,8 @@ def train(args):
     training_details = client.training.get_details(run_uid)
     print("training_details", training_details)
 
-    training_uid = training_details['entity']['results_reference']['location']['training']
     Path(args.output_training_uid_path).parent.mkdir(parents=True, exist_ok=True)
-    Path(args.output_training_uid_path).write_text(training_uid)
+    Path(args.output_training_uid_path).write_text(run_uid)
 
 if __name__ == "__main__":
     import argparse