import PythonKit or Python (#393)

marcrasi · web-flow · commit 1db911dbe976 · 2020-02-25T20:53:03.000-08:00
diff --git a/docs/site/tutorials/model_training_walkthrough.ipynb b/docs/site/tutorials/model_training_walkthrough.ipynb
@@ -128,7 +128,11 @@
    "outputs": [],
    "source": [
     "import TensorFlow\n",
-    "import Python"
+    "#if canImport(PythonKit)\n",
+    "    import PythonKit\n",
+    "#else\n",
+    "    import Python\n",
+    "#endif"
    ]
   },
   {
diff --git a/docs/site/tutorials/python_interoperability.ipynb b/docs/site/tutorials/python_interoperability.ipynb
@@ -78,7 +78,11 @@
    },
    "outputs": [],
    "source": [
-    "import Python\n",
+    "#if canImport(PythonKit)\n",
+    "    import PythonKit\n",
+    "#else\n",
+    "    import Python\n",
+    "#endif\n",
     "print(Python.version)"
    ]
   },
