Make enemble's object and source fields public

lincc-frameworks · Sep 18, 2023 · 5621a0b · 5621a0b
1 parent 7561df3
commit 5621a0b
Show file tree

Hide file tree

Showing 5 changed files with 1,501 additions and 204 deletions.
diff --git a/docs/tutorials/binning_slowly_changing_sources.ipynb b/docs/tutorials/binning_slowly_changing_sources.ipynb
@@ -60,7 +60,7 @@
    "outputs": [],
    "source": [
     "fig, ax = plt.subplots(1, 1)\n",
-    "_ = ax.hist(ens._source[\"midPointTai\"].compute().tolist(), 500)\n",
+    "_ = ax.hist(ens.source[\"midPointTai\"].compute().tolist(), 500)\n",
     "_ = ax.set_xlabel(\"Time (MJD)\")\n",
     "_ = ax.set_ylabel(\"Source Count\")"
    ]
@@ -90,7 +90,7 @@
    "source": [
     "ens.bin_sources(time_window=7.0, offset=0.0)\n",
     "fig, ax = plt.subplots(1, 1)\n",
-    "_ = ax.hist(ens._source[\"midPointTai\"].compute().tolist(), 500)\n",
+    "_ = ax.hist(ens.source[\"midPointTai\"].compute().tolist(), 500)\n",
     "_ = ax.set_xlabel(\"Time (MJD)\")\n",
     "_ = ax.set_ylabel(\"Source Count\")"
    ]
@@ -120,7 +120,7 @@
    "source": [
     "ens.bin_sources(time_window=28.0, offset=0.0, custom_aggr={\"midPointTai\": \"min\"})\n",
     "fig, ax = plt.subplots(1, 1)\n",
-    "_ = ax.hist(ens._source[\"midPointTai\"].compute().tolist(), 500)\n",
+    "_ = ax.hist(ens.source[\"midPointTai\"].compute().tolist(), 500)\n",
     "_ = ax.set_xlabel(\"Time (MJD)\")\n",
     "_ = ax.set_ylabel(\"Source Count\")"
    ]
@@ -150,7 +150,7 @@
     "ens.from_source_dict(rows, column_mapper=cmap)\n",
     "\n",
     "fig, ax = plt.subplots(1, 1)\n",
-    "_ = ax.hist(ens._source[\"midPointTai\"].compute().tolist(), 60)\n",
+    "_ = ax.hist(ens.source[\"midPointTai\"].compute().tolist(), 60)\n",
     "_ = ax.set_xlabel(\"Time (MJD)\")\n",
     "_ = ax.set_ylabel(\"Source Count\")"
    ]
@@ -179,7 +179,7 @@
     "ens.bin_sources(time_window=1.0, offset=0.0)\n",
     "\n",
     "fig, ax = plt.subplots(1, 1)\n",
-    "_ = ax.hist(ens._source[\"midPointTai\"].compute().tolist(), 60)\n",
+    "_ = ax.hist(ens.source[\"midPointTai\"].compute().tolist(), 60)\n",
     "_ = ax.set_xlabel(\"Time (MJD)\")\n",
     "_ = ax.set_ylabel(\"Source Count\")"
    ]
@@ -209,7 +209,7 @@
     "ens.bin_sources(time_window=1.0, offset=0.5)\n",
     "\n",
     "fig, ax = plt.subplots(1, 1)\n",
-    "_ = ax.hist(ens._source[\"midPointTai\"].compute().tolist(), 60)\n",
+    "_ = ax.hist(ens.source[\"midPointTai\"].compute().tolist(), 60)\n",
     "_ = ax.set_xlabel(\"Time (MJD)\")\n",
     "_ = ax.set_ylabel(\"Source Count\")"
    ]
@@ -259,7 +259,7 @@
     "ens.bin_sources(time_window=1.0, offset=0.5)\n",
     "\n",
     "fig, ax = plt.subplots(1, 1)\n",
-    "_ = ax.hist(ens._source[\"midPointTai\"].compute().tolist(), 60)\n",
+    "_ = ax.hist(ens.source[\"midPointTai\"].compute().tolist(), 60)\n",
     "_ = ax.set_xlabel(\"Time (MJD)\")\n",
     "_ = ax.set_ylabel(\"Source Count\")"
    ]

diff --git a/docs/tutorials/scaling_to_large_data.ipynb b/docs/tutorials/scaling_to_large_data.ipynb
@@ -216,7 +216,7 @@
     "\n",
     "print(\"number of lightcurve results in mapres: \", len(mapres))\n",
     "print(\"number of lightcurve results in groupres: \", len(groupres))\n",
-    "print(\"True number of lightcurves in the dataset:\", len(np.unique(ens._source.index)))"
+    "print(\"True number of lightcurves in the dataset:\", len(np.unique(ens.source.index)))"
    ]
   },
   {