Test alternate cmfgen path

tardis-sn · Apr 12, 2024 · d3c8e9d · d3c8e9d
1 parent 7588efd
commit d3c8e9d
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/docs/tardis_atomdata_ref.ipynb b/docs/tardis_atomdata_ref.ipynb
@@ -19,6 +19,7 @@
    },
    "outputs": [],
    "source": [
+    "import pathlib\n",
     "from carsus.io.nist import NISTWeightsComp, NISTIonizationEnergies"
    ]
   },
@@ -59,7 +60,9 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "cmfgen_path = '../../carsus-data-cmfgen/atomic/'"
+    "cmfgen_path = '../../carsus-data-cmfgen/atomic/'\n",
+    "if not pathlib.Path(cmfgen_path).exists():\n",
+    "    cmfgen_path = \"/tmp/atomic/\""
    ]
   },
   {