Remove TFX version pin to fix import error

PiperOrigin-RevId: 452375524
tensorflow · Jun 1, 2022 · 32168c6 · 32168c6
1 parent e797ae5
commit 32168c6
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/docs/tutorials/tfx/neural_structured_learning.ipynb b/docs/tutorials/tfx/neural_structured_learning.ipynb
@@ -142,11 +142,9 @@
       },
       "outputs": [],
       "source": [
-        "try:\n",
-        "  import colab\n",
-        "  !pip install --upgrade pip\n",
-        "except:\n",
-        "  pass"
+        "import sys\n",
+        "if 'google.colab' in sys.modules:\n",
+        "  !pip install --upgrade pip"
       ]
     },
     {
@@ -167,7 +165,7 @@
       "outputs": [],
       "source": [
         "!pip install -q -U \\\n",
-        "  tfx==1.2.0 \\\n",
+        "  tfx \\\n",
         "  neural-structured-learning \\\n",
         "  tensorflow-hub \\\n",
         "  tensorflow-datasets"