Copy over updates to CMNN demo in rail

LSSTDESC · Nov 14, 2023 · 653c00c · 653c00c
1 parent f2eae3c
commit 653c00c
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/examples/estimation_examples/CMNN_Demo.ipynb b/examples/estimation_examples/CMNN_Demo.ipynb
@@ -4,7 +4,7 @@
    "cell_type": "markdown",
    "metadata": {},
    "source": [
-    "# CMNN Demo\n",
+    "# RAIL CMNN Tutorial Notebook\n",
     "\n",
     "**Author:** Sam Schmidt\n",
     "\n",
@@ -135,9 +135,9 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "from rail.core.utils import find_rail_file\n",
-    "trainFile = find_rail_file('examples_data/testdata/test_dc2_training_9816.hdf5')\n",
-    "testFile = find_rail_file('examples_data/testdata/test_dc2_validation_9816.hdf5')\n",
+    "from rail.core.utils import RAILDIR\n",
+    "trainFile = os.path.join(RAILDIR, 'rail/examples_data/testdata/test_dc2_training_9816.hdf5')\n",
+    "testFile = os.path.join(RAILDIR, 'rail/examples_data/testdata/test_dc2_validation_9816.hdf5')\n",
     "training_data = DS.read_file(\"training_data\", TableHandle, trainFile)\n",
     "test_data = DS.read_file(\"test_data\", TableHandle, testFile)"
    ]