Jupyterbook docs fixes 2 (#9)

* adapt pyg writing for neural-lam

* fix missing and circular imports

* make regular input grid assumption explicit

* handle missing edge attributes in splitting

* minor fix for neural-lam save

* add missing dep

* tweak cicd

.github/workflows/deploy-docs-book.yml

@@ -24,7 +24,9 @@ jobs:
         cache: true
 
     - name: Install dependencies
-      run: pdm install --group docs
+      run: |
+        pdm install --prod
+        pdm install --group docs
 
     # Build the book
     - name: Build the book

pyproject.toml

@@ -25,6 +25,7 @@ visualisation = [
 ]
 docs = [
     "jupyter-book>=1.0.0",
+    "sphinxcontrib-mermaid>=0.9.2",
 ]
 
 [build-system]

src/weather_model_graphs/networkx_utils.py

@@ -103,6 +103,17 @@ def split_graph_by_edge_attribute(graph, attr):
             f"No subgraphs were created. Check the edge attribute '{attr}'."
         )
 
+    # copy node attributes
+    for subgraph in subgraphs.values():
+        for node in subgraph.nodes:
+            subgraph.nodes[node].update(graph.nodes[node])
+
+    # check that at least one subgraph was created
+    if len(subgraphs) == 0:
+        raise ValueError(
+            f"No subgraphs were created. Check the edge attribute '{attr}'."
+        )
+
     return subgraphs