update

docs/source/dataset_implements.ipynb

@@ -20,7 +20,7 @@
     "from pyspark.sql import SparkSession\n",
     "\n",
     "spark = SparkSession.Builder().config(\"spark.ui.showConsoleProgress\", \"false\").getOrCreate()\n",
-    "spark.sparkContext.setLogLevel(\"ERROR\")"
+    "spark.sparkContext.setLogLevel(\"ERROR\")\n"
    ]
   },
   {
@@ -56,7 +56,7 @@
     "\n",
     "person = create_empty_dataset(spark, Person)\n",
     "pet = create_empty_dataset(spark, Pet)\n",
-    "fruit = create_empty_dataset(spark, Fruit)"
+    "fruit = create_empty_dataset(spark, Fruit)\n"
    ]
   },
   {
@@ -86,7 +86,7 @@
     "        df,\n",
     "        df.typedspark_schema,\n",
     "        {Person.age: Person.age + 1},\n",
-    "    )\n"
+    "    )"
    ]
   },
   {
@@ -123,7 +123,7 @@
     "        df,\n",
     "        df.typedspark_schema,\n",
     "        {Age.age: Age.age + 1},\n",
-    "    )\n"
+    "    )"
    ]
   },
   {
@@ -147,15 +147,19 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "happy_person = birthday(person)  # returns a DataSet[Person]\n",
-    "happy_pet = birthday(pet)  # returns a DataSet[Pet]\n",
+    "# returns a DataSet[Person]\n",
+    "happy_person = birthday(person)\n",
+    "\n",
+    "# returns a DataSet[Pet]\n",
+    "happy_pet = birthday(pet)\n",
+    "\n",
     "try:\n",
     "    # Raises a linting error:\n",
     "    # Argument of type \"DataSet[Fruit]\" cannot be assigned to\n",
     "    # parameter \"df\" of type \"DataSetImplements[Age, T@birthday]\"\n",
     "    birthday(fruit)\n",
     "except Exception as e:\n",
-    "    pass"
+    "    pass\n"
    ]
   },
   {