Remove expectation to get RailStage in import_and_attach_all

examples/core_examples/Useful_Utilities.ipynb

@@ -75,7 +75,7 @@
    "source": [
     "import rail\n",
     "import rail.stages\n",
-    "rail.stages.import_and_attach_all()\n"
+    "rail.stages.import_and_attach_all()"
    ]
   },
   {
@@ -136,9 +136,7 @@
    "source": [
     "### Listing imported stages (2/2)\n",
     "\n",
-    "Now, let's try listing imported stages again.\n",
-    "\n",
-    "Note that we can now just call `stage.RailStage` instead of `rail.core.stage.RailStage`."
+    "Now, let's try listing imported stages again, and notice how many more we get."
    ]
   },
   {
@@ -147,7 +145,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "for val in stage.RailStage.pipeline_stages.values():\n",
+    "for val in rail.core.stage.RailStage.pipeline_stages.values():\n",
     "    print(val[0])"
    ]
   },
@@ -166,7 +164,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "for val in stage.RailStage.pipeline_stages.values():\n",
+    "for val in rail.core.stage.RailStage.pipeline_stages.values():\n",
     "    if issubclass(val[0], rail.estimation.estimator.CatEstimator):\n",
     "        print(val[0])"
    ]
@@ -188,7 +186,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "DS = stage.RailStage.data_store\n",
+    "DS = rail.core.stage.RailStage.data_store\n",
     "DS.__class__.allow_overwrite = True"
    ]
   },
@@ -278,11 +276,12 @@
    "outputs": [],
    "source": [
     "# Now, we have to set up some other variables for our pipeline:\n",
+    "import numpy as np\n",
     "\n",
     "bands = [\"u\", \"g\", \"r\", \"i\", \"z\", \"y\"]\n",
     "band_dict = {band: f\"mag_{band}_lsst\" for band in bands}\n",
     "rename_dict = {f\"mag_{band}_lsst_err\": f\"mag_err_{band}_lsst\" for band in bands}\n",
-    "post_grid = [float(x) for x in np.linspace(0.0, 5, 21)]\n"
+    "post_grid = [float(x) for x in np.linspace(0.0, 5, 21)]"
    ]
   },
   {
@@ -585,7 +584,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.10.0"
+   "version": "3.11.6"
   }
  },
  "nbformat": 4,