Merge pull request #275 from openclimatefix/fix/top-import

fix trying to import OpenTopography
openclimatefix · Feb 23, 2024 · 4ae78d1 · 4ae78d1
2 parents 64ed0e0 + 8d25def
commit 4ae78d1
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/.github/workflows/workflows.yaml b/.github/workflows/workflows.yaml
@@ -16,4 +16,4 @@ jobs:
       sudo_apt_install: "libgeos++-dev libproj-dev proj-data proj-bin"
       #      brew_install: "proj geos librttopo"
       os_list: '["ubuntu-latest"]'
-      python-version: "['3.9','3.10','3.11']"
+      python-version: "['3.10','3.11']"
diff --git a/ocf_datapipes/training/common.py b/ocf_datapipes/training/common.py
@@ -19,11 +19,21 @@
     OpenPVFromNetCDF,
     OpenPVFromPVSitesDB,
     OpenSatellite,
-    OpenTopography,
     OpenWindFromNetCDF,
 )
 from ocf_datapipes.utils.utils import flatten_nwp_source_dict
 
+try:
+    from ocf_datapipes.load import OpenTopography
+
+    # Rioxarray is sometimes a pain to install, so only load this if its installed
+except ImportError:
+    print(
+        "Could not import OpenTopography," " this is probably becasye Rioxarray is not installed."
+    )
+    pass
+
+
 logger = logging.getLogger(__name__)