From bdc1a1472a6a385a4749947c9d9306e258add256 Mon Sep 17 00:00:00 2001
From: Moka-4444 <159596295+Moka-4444@users.noreply.github.com>
Date: Mon, 23 Mar 2026 16:49:54 +0200
Subject: [PATCH] Fix #95: Unpack node_features_values when list_from_attribute
 is None

Resolves an issue in save.py where node_features_values was returning a list instead of a tensor.
---
 src/weather_model_graphs/save.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/weather_model_graphs/save.py b/src/weather_model_graphs/save.py
index 74ff5ef..34877f1 100644
--- a/src/weather_model_graphs/save.py
+++ b/src/weather_model_graphs/save.py
@@ -142,6 +142,7 @@ def _concat_pyg_features(
     if list_from_attribute is None:
         edge_features_values = edge_features_values[0]
         edge_indecies = edge_indecies[0]
+                node_features_values = node_features_values[0]
 
     Path(output_directory).mkdir(exist_ok=True, parents=True)
     fp_edge_index = Path(output_directory) / f"{name}_edge_index.pt"