From 3c54ff73de18bb5ec2204ba7fea389cdf40227f3 Mon Sep 17 00:00:00 2001
From: Doug Branton <brantd@uw.edu>
Date: Wed, 10 Jan 2024 11:07:49 -0800
Subject: [PATCH] add save docs

---
 .../tutorials/working_with_the_ensemble.ipynb | 39 +++++++++----------
 1 file changed, 18 insertions(+), 21 deletions(-)

diff --git a/docs/tutorials/working_with_the_ensemble.ipynb b/docs/tutorials/working_with_the_ensemble.ipynb
index a0f5b2ec..25a4818e 100644
--- a/docs/tutorials/working_with_the_ensemble.ipynb
+++ b/docs/tutorials/working_with_the_ensemble.ipynb
@@ -26,7 +26,7 @@
     "import numpy as np\n",
     "import pandas as pd\n",
     "\n",
-    "np.random.seed(1)    \n",
+    "np.random.seed(1)\n",
     "\n",
     "# Generate 10 astronomical objects\n",
     "n_obj = 10\n",
@@ -34,9 +34,9 @@
     "names = ids.astype(str)\n",
     "object_table = pd.DataFrame(\n",
     "    {\n",
-    "        \"id\": ids, \n",
+    "        \"id\": ids,\n",
     "        \"name\": names,\n",
-    "        \"ddf_bool\": np.random.randint(0, 2, n_obj), # 0 if from deep drilling field, 1 otherwise\n",
+    "        \"ddf_bool\": np.random.randint(0, 2, n_obj),  # 0 if from deep drilling field, 1 otherwise\n",
     "        \"libid_cadence\": np.random.randint(1, 130, n_obj),\n",
     "    }\n",
     ")\n",
@@ -49,7 +49,7 @@
     "    {\n",
     "        \"id\": 8000 + (np.arange(num_points) % n_obj),\n",
     "        \"time\": np.arange(num_points),\n",
-    "        \"flux\":  np.random.random_sample(size=num_points)*10,\n",
+    "        \"flux\": np.random.random_sample(size=num_points) * 10,\n",
     "        \"band\": np.repeat(all_bands, num_points / len(all_bands)),\n",
     "        \"error\": np.random.random_sample(size=num_points),\n",
     "        \"count\": np.arange(num_points),\n",
@@ -89,7 +89,8 @@
     "    flux_col=\"flux\",\n",
     "    err_col=\"error\",\n",
     "    band_col=\"band\",\n",
-    "    npartitions=1)"
+    "    npartitions=1,\n",
+    ")"
    ]
   },
   {
@@ -124,18 +125,12 @@
     "from tape.utils import ColumnMapper\n",
     "\n",
     "# columns assigned manually\n",
-    "col_map = ColumnMapper().assign(id_col=\"id\",\n",
-    "                                time_col=\"time\",\n",
-    "                                flux_col=\"flux\",\n",
-    "                                err_col=\"error\",\n",
-    "                                band_col=\"band\")\n",
+    "col_map = ColumnMapper().assign(\n",
+    "    id_col=\"id\", time_col=\"time\", flux_col=\"flux\", err_col=\"error\", band_col=\"band\"\n",
+    ")\n",
     "\n",
     "# Pass the ColumnMapper along to from_pandas\n",
-    "ens.from_pandas(\n",
-    "    source_frame=source_table,\n",
-    "    object_frame=object_table,\n",
-    "    column_mapper=col_map,\n",
-    "    npartitions=1)"
+    "ens.from_pandas(source_frame=source_table, object_frame=object_table, column_mapper=col_map, npartitions=1)"
    ]
   },
   {
@@ -616,8 +611,8 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "ens.add_frame(ens.select_frame(\"stetson_j\"), \"stetson_j_result_1\") # Add result under new label\n",
-    "ens.drop_frame(\"stetson_j\") # Drop original label\n",
+    "ens.add_frame(ens.select_frame(\"stetson_j\"), \"stetson_j_result_1\")  # Add result under new label\n",
+    "ens.drop_frame(\"stetson_j\")  # Drop original label\n",
     "\n",
     "ens.select_frame(\"stetson_j_result_1\").compute()"
    ]
@@ -655,7 +650,7 @@
     "ens.drop_frame(\"result_1\")\n",
     "\n",
     "try:\n",
-    "    ens.select_frame(\"result_1\") # This should result in a KeyError since the frame has been dropped.\n",
+    "    ens.select_frame(\"result_1\")  # This should result in a KeyError since the frame has been dropped.\n",
     "except Exception as e:\n",
     "    print(\"As expected, the frame 'result_1 was dropped.\\n\" + str(e))"
    ]
@@ -842,8 +837,10 @@
    "outputs": [],
    "source": [
     "ens.save_ensemble(\n",
-    "    \".\", \"ensemble\", additional_frames=[\"result_3\"],\n",
-    "    ) # Saves object, source, and result_3 to disk"
+    "    \".\",\n",
+    "    \"ensemble\",\n",
+    "    additional_frames=[\"result_3\"],\n",
+    ")  # Saves object, source, and result_3 to disk"
    ]
   },
   {
@@ -862,7 +859,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "new_ens = Ensemble(client=ens.client) # use the same client\n",
+    "new_ens = Ensemble(client=ens.client)  # use the same client\n",
     "new_ens.from_ensemble(\"./ensemble\", additional_frames=True)\n",
     "new_ens.select_frame(\"result_3\").head(5)"
    ]