Update missing notebook

lincc-frameworks · Sep 18, 2023 · f6e1f4d · f6e1f4d
1 parent 5621a0b
commit f6e1f4d
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/docs/tutorials/using_ray_with_the_ensemble.ipynb b/docs/tutorials/using_ray_with_the_ensemble.ipynb
@@ -81,7 +81,7 @@
    "outputs": [],
    "source": [
     "ens.from_dataset(\"s82_qso\")\n",
-    "ens._source = ens._source.repartition(npartitions=10)\n",
+    "ens.source = ens.source.repartition(npartitions=10)\n",
     "ens.batch(calc_sf2, use_map=False)  # use_map is false as we repartition naively, splitting per-object sources across partitions"
    ]
   },
@@ -116,7 +116,7 @@
     "\n",
     "ens=Ensemble(client=False) # Do not use a client\n",
     "ens.from_dataset(\"s82_qso\")\n",
-    "ens._source = ens._source.repartition(npartitions=10)\n",
+    "ens.source = ens.source.repartition(npartitions=10)\n",
     "ens.batch(calc_sf2, use_map=False)"
    ]
   },
@@ -150,7 +150,7 @@
     "\n",
     "ens = Ensemble()\n",
     "ens.from_dataset(\"s82_qso\")\n",
-    "ens._source = ens._source.repartition(npartitions=10)\n",
+    "ens.source = ens.source.repartition(npartitions=10)\n",
     "ens.batch(calc_sf2, use_map=False)"
    ]
   }