From 73f1ca2a77e2ca0a57400712156465d6172b9f1a Mon Sep 17 00:00:00 2001 From: Mario Santa Cruz Date: Thu, 8 Aug 2024 16:02:07 +0000 Subject: [PATCH] Update error messages from invalid sub_graph in model instantiation --- CHANGELOG.md | 2 ++ src/anemoi/models/layers/mapper.py | 4 ++-- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/CHANGELOG.md b/CHANGELOG.md index d877b34..84a4cbf 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -16,6 +16,8 @@ Keep it human-readable, your future self will thank you! ### Changed +- New error messages for wrongs graphs. + ### Removed ## 0.2.0 diff --git a/src/anemoi/models/layers/mapper.py b/src/anemoi/models/layers/mapper.py index 04efdf0..0967041 100644 --- a/src/anemoi/models/layers/mapper.py +++ b/src/anemoi/models/layers/mapper.py @@ -134,8 +134,8 @@ def _register_edges( trainable_size : int Trainable tensor size """ - if edge_attributes is None: - raise ValueError("Edge attributes must be provided") + assert sub_graph, f"{self.__class__.__name__} needs a valid sub_graph to register edges." + assert edge_attributes is not None, "Edge attributes must be provided" edge_attr_tensor = torch.cat([sub_graph[attr] for attr in edge_attributes], axis=1)