update to kubeflow-metadata in examples (kubeflow#646)

AmandeepSinghCS · Sep 26, 2019 · cfe166f · cfe166f
1 parent 628babc
commit cfe166f
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 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
@@ -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
@@ -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
@@ -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
@@ -2,7 +2,7 @@ fire
 gitpython
 google-cloud-storage
 joblib
-kfmd
+kubeflow-metadata
 numpy
 pandas
 retrying