Minor fixes to visuals

ecmwf · Sep 20, 2024 · 1fd8711 · 1fd8711
1 parent 14097e2
commit 1fd8711
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 181,523 deletions.
diff --git a/example_graph_config.yaml b/example_graph_config.yaml
@@ -3,7 +3,6 @@ creator: HierarchicalGraphCreator
 
 num_hidden: 3
 level_process: True
-data: 'data'
 
 processor_node: 
   _target_: anemoi.graphs.nodes.TriNodes

diff --git a/src/anemoi/graphs/plotting/graph_vis_funcs.py b/src/anemoi/graphs/plotting/graph_vis_funcs.py
@@ -201,7 +201,7 @@ def vis_downscale(data_nodes, hidden_nodes, data_to_hidden_edges, downscale_edge
 
     colorscale = generate_shades(color, num_hidden)
     layout = make_layout(title)
-    scale_increment = 1/num_hidden
+    scale_increment = 1/(num_hidden+1)
 
     # Data
     g_data = to_networkx(
@@ -267,7 +267,7 @@ def vis_upscale(data_nodes, hidden_nodes, data_to_hidden_edges, upscale_edges, t
 
     colorscale = generate_shades(color, num_hidden)
     layout = make_layout(title)
-    scale_increment = 1/num_hidden
+    scale_increment = 1/(num_hidden+1)
 
     # Hidden
     graphs = []
@@ -336,7 +336,7 @@ def vis_level(data_nodes, hidden_nodes, data_to_hidden_edges, hidden_edges, titl
 
     colorscale = generate_shades(color, num_hidden)
     layout = make_layout(title)
-    scale_increment = 1/num_hidden
+    scale_increment = 1/(num_hidden+1)
 
 
     # Data

diff --git a/src/anemoi/graphs/plotting/hierarchical_dynamic_vis.ipynb b/src/anemoi/graphs/plotting/hierarchical_dynamic_vis.ipynb