Fix notebooks

tests/all-spark-notebook/data/issue_1168.ipynb

@@ -9,48 +9,37 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 6,
+   "execution_count": null,
    "metadata": {},
    "outputs": [],
    "source": [
     "from pyspark.sql import SparkSession\n",
-    "from pyspark.sql.functions import pandas_udf\n",
     "\n",
     "# Spark session & context\n",
-    "spark = SparkSession.builder.master('local').getOrCreate()"
+    "spark = SparkSession.builder.master(\"local\").getOrCreate()"
    ]
   },
   {
    "cell_type": "code",
-   "execution_count": 7,
+   "execution_count": null,
    "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "+---+---+\n",
-      "| id|age|\n",
-      "+---+---+\n",
-      "|  1| 21|\n",
-      "+---+---+\n",
-      "\n"
-     ]
-    }
-   ],
+   "outputs": [],
    "source": [
     "df = spark.createDataFrame([(1, 21), (2, 30)], (\"id\", \"age\"))\n",
+    "\n",
+    "\n",
     "def filter_func(iterator):\n",
     "    for pdf in iterator:\n",
     "        yield pdf[pdf.id == 1]\n",
     "\n",
+    "\n",
     "df.mapInPandas(filter_func, df.schema).show()"
    ]
   }
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "Python 3",
+   "display_name": "Python 3 (ipykernel)",
    "language": "python",
    "name": "python3"
   },
@@ -64,7 +53,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.8.6"
+   "version": "3.9.10"
   }
  },
  "nbformat": 4,