From cfe166f73fa2903f83d540fd7cf4045d9de7fa03 Mon Sep 17 00:00:00 2001
From: Jin Chi He <hejinchi@cn.ibm.com>
Date: Fri, 27 Sep 2019 07:13:34 +0800
Subject: [PATCH] update to kubeflow-metadata in examples (#646)

---
 .../components/t2t/containers/metadata-logger/Dockerfile   | 2 +-
 .../components/t2t/metadata-logger/log-metadata.py         | 2 +-
 xgboost_synthetic/build-train-deploy.ipynb                 | 7 +++----
 xgboost_synthetic/requirements.txt                         | 2 +-
 4 files changed, 6 insertions(+), 7 deletions(-)

diff --git a/github_issue_summarization/pipelines/components/t2t/containers/metadata-logger/Dockerfile b/github_issue_summarization/pipelines/components/t2t/containers/metadata-logger/Dockerfile
index 3e7f6fb84..fd1e396ee 100644
--- a/github_issue_summarization/pipelines/components/t2t/containers/metadata-logger/Dockerfile
+++ b/github_issue_summarization/pipelines/components/t2t/containers/metadata-logger/Dockerfile
@@ -26,7 +26,7 @@ RUN apt-get install -y wget unzip git
 # RUN pip install tensorflow==1.12.0
 
 RUN pip install --upgrade pip
-RUN pip install kfmd urllib3 certifi retrying
+RUN pip install kubeflow-metadata urllib3 certifi retrying
 
 # RUN wget -nv https://dl.google.com/dl/cloudsdk/release/google-cloud-sdk.zip && \
 #     unzip -qq google-cloud-sdk.zip -d tools && \
diff --git a/github_issue_summarization/pipelines/components/t2t/metadata-logger/log-metadata.py b/github_issue_summarization/pipelines/components/t2t/metadata-logger/log-metadata.py
index 05ce3b991..f3e504c88 100644
--- a/github_issue_summarization/pipelines/components/t2t/metadata-logger/log-metadata.py
+++ b/github_issue_summarization/pipelines/components/t2t/metadata-logger/log-metadata.py
@@ -17,7 +17,7 @@
 import logging
 import retrying
 
-from kfmd import metadata
+from kubeflow.metadata import metadata #pylint: disable=no-name-in-module
 
 DATASET = 'dataset'
 MODEL = 'model'
diff --git a/xgboost_synthetic/build-train-deploy.ipynb b/xgboost_synthetic/build-train-deploy.ipynb
index 7806978a8..7ae63f2ea 100644
--- a/xgboost_synthetic/build-train-deploy.ipynb
+++ b/xgboost_synthetic/build-train-deploy.ipynb
@@ -119,7 +119,6 @@
     "import fire\n",
     "import joblib\n",
     "import logging\n",
-    "import kfmd\n",
     "import nbconvert\n",
     "import os\n",
     "import pathlib\n",
@@ -133,9 +132,9 @@
     "from xgboost import XGBRegressor\n",
     "from importlib import reload\n",
     "from sklearn.datasets import make_regression\n",
-    "from kfmd import metadata\n",
-    "from kfmd import openapi_client\n",
-    "from kfmd.openapi_client import Configuration, ApiClient, MetadataServiceApi\n",
+    "from kubeflow.metadata import metadata\n",
+    "from kubeflow.metadata import openapi_client\n",
+    "from kubeflow.metadata.openapi_client import Configuration, ApiClient, MetadataServiceApi\n",
     "from datetime import datetime\n",
     "import retrying\n",
     "import urllib3"
diff --git a/xgboost_synthetic/requirements.txt b/xgboost_synthetic/requirements.txt
index 32821f169..7e3ae1de2 100644
--- a/xgboost_synthetic/requirements.txt
+++ b/xgboost_synthetic/requirements.txt
@@ -2,7 +2,7 @@ fire
 gitpython
 google-cloud-storage
 joblib
-kfmd
+kubeflow-metadata
 numpy
 pandas
 retrying