MCT 1.2 Release

PiperOrigin-RevId: 419611222
tensorflow · Jan 4, 2022 · 56f8092 · 56f8092
1 parent 2817be9
commit 56f8092
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 4 deletions.
diff --git a/model_card_toolkit/documentation/examples/MLMD_Model_Card_Toolkit_Demo.ipynb b/model_card_toolkit/documentation/examples/MLMD_Model_Card_Toolkit_Demo.ipynb
@@ -102,7 +102,6 @@
       "outputs": [],
       "source": [
         "!pip install --upgrade pip==20.2\n",
-        "!pip install \"tfx==1.2.0\"\n",
         "!pip install model-card-toolkit"
       ]
     },
@@ -1080,7 +1079,8 @@
       "source": [
         "import model_card_toolkit as mctlib\n",
         "\n",
-        "mct = mctlib.ModelCardToolkit(mlmd_store=mlmd_store, model_uri=model_uri)"
+        "mlmd_source = mctlib.utils.MlmdSource(store=mlmd_store, model_uri=model_uri)\n",
+        "mct = mctlib.ModelCardToolkit(mlmd_source=mlmd_source)"
       ]
     },
     {
@@ -1281,7 +1281,7 @@
       },
       "outputs": [],
       "source": [
-        "mct.update_model_card_json(model_card)"
+        "mct.update_model_card(model_card)"
       ]
     },
     {

diff --git a/model_card_toolkit/version.py b/model_card_toolkit/version.py
@@ -14,4 +14,4 @@
 """Contains the version string of the Model Card Toolkit."""
 
 # Note that setup.py uses this version.
-__version__ = '1.1.1.dev'
+__version__ = '1.2.0'
diff --git a/setup.py b/setup.py
@@ -36,6 +36,7 @@
     'jsonschema>=3.2.0,<4',
     'tensorflow-model-analysis>=0.36.0,<0.37.0',
     'tensorflow-metadata>=1.5.0,<1.6.0',
+    'tfx>=1.5.0,<1.6.0',
     'ml-metadata>=1.5.0,<1.6.0',
     'dataclasses;python_version<"3.7"',
 ]