updated for new directories

LSSTDESC · May 4, 2023 · 6c93b63 · 6c93b63
1 parent bae9d2a
commit 6c93b63
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/nb/examples/goldenspike/goldenspike.ipynb b/nb/examples/goldenspike/goldenspike.ipynb
@@ -39,7 +39,7 @@
    "outputs": [],
    "source": [
     "from rail.core.utils import RAILDIR\n",
-    "flow_file = os.path.join(RAILDIR, 'rail/examples/goldenspike/data/pretrained_flow.pkl')\n",
+    "flow_file = os.path.join(RAILDIR, 'rail/examples_data/goldenspike_data/data/pretrained_flow.pkl')\n",
     "pipe.initialize(dict(model=flow_file), dict(output_dir='.', log_dir='.', resume=False), None)"
    ]
   },
@@ -119,7 +119,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.9.13"
+   "version": "3.10.10"
   }
  },
  "nbformat": 4,

diff --git a/src/rail/pipelines/examples/goldenspike/goldenspike.py b/src/rail/pipelines/examples/goldenspike/goldenspike.py
@@ -16,7 +16,7 @@
 
 namer = NameFactory()
 from rail.core.utils import RAILDIR
-flow_file = os.path.join(RAILDIR, 'rail/examples/goldenspike/data/pretrained_flow.pkl')
+flow_file = os.path.join(RAILDIR, 'rail/examples_data/goldenspike_data/data/pretrained_flow.pkl')
 
 
 class GoldenspikePipeline(RailPipeline):

diff --git a/tests/test_pipelines.py b/tests/test_pipelines.py
@@ -7,7 +7,7 @@
 
 def test_golden():
     namer = NameFactory()
-    flow_file = os.path.join(RAILDIR, 'examples/goldenspike/data/pretrained_flow.pkl')
+    flow_file = os.path.join(RAILDIR, 'rail/examples_data/goldenspike_data/data/pretrained_flow.pkl')
     output_dir = namer.get_project_dir('.', 'tmp_test', 'tmp_test')
     try:
         os.makedirs(output_dir)