Add stage prefix to RailStage (even after import and attach all)

LSSTDESC · Jul 8, 2024 · 93087f9 · 93087f9
1 parent fe64556
commit 93087f9
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/examples/core_examples/Useful_Utilities.ipynb b/examples/core_examples/Useful_Utilities.ipynb
@@ -138,7 +138,7 @@
     "\n",
     "Now, let's try listing imported stages again.\n",
     "\n",
-    "Note that we can now just call `RailStage` instead of `rail.core.stage.RailStage`."
+    "Note that we can now just call `stage.RailStage` instead of `rail.core.stage.RailStage`."
    ]
   },
   {
@@ -147,7 +147,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "for val in RailStage.pipeline_stages.values():\n",
+    "for val in stage.RailStage.pipeline_stages.values():\n",
     "    print(val[0])"
    ]
   },
@@ -166,7 +166,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "for val in RailStage.pipeline_stages.values():\n",
+    "for val in stage.RailStage.pipeline_stages.values():\n",
     "    if issubclass(val[0], rail.estimation.estimator.CatEstimator):\n",
     "        print(val[0])"
    ]
@@ -188,7 +188,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "DS = RailStage.data_store\n",
+    "DS = stage.RailStage.data_store\n",
     "DS.__class__.allow_overwrite = True"
    ]
   },