Update dataloading import to new neuralhydrology version

gauchm · Nov 13, 2020 · 11c0484 · 11c0484
1 parent 316cdc9
commit 11c0484
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/results-analysis.ipynb b/results-analysis.ipynb
@@ -36,7 +36,11 @@
     "from scipy.stats import wilcoxon, pearsonr\n",
     "from tqdm.notebook import tqdm\n",
     "\n",
-    "from neuralhydrology.data.utils import load_camels_us_forcings, load_hourly_us_netcdf, load_camels_us_attributes\n",
+    "# changed in a newer version of neuralhydrology\n",
+    "try:\n",
+    "    from neuralhydrology.data.utils import load_camels_us_forcings, load_hourly_us_netcdf, load_camels_us_attributes\n",
+    "except ModuleNotFoundError:\n",
+    "    from neuralhydrology.datasetzoo.camelsus import load_camels_us_forcings, load_camels_us_attributes\n",
     "from neuralhydrology.evaluation.metrics import calculate_metrics\n",
     "from neuralhydrology.evaluation.signatures import calculate_signatures, get_available_signatures\n",
     "\n",