Skip to content
This repository has been archived by the owner on Dec 20, 2024. It is now read-only.

Commit

Permalink
fix: assert missing for Hexagonal edges
Browse files Browse the repository at this point in the history
  • Loading branch information
theissenhelen committed Jul 2, 2024
1 parent 2e6830f commit a59f5d1
Showing 1 changed file with 3 additions and 0 deletions.
3 changes: 3 additions & 0 deletions src/anemoi/graphs/edges/builder.py
Original file line number Diff line number Diff line change
Expand Up @@ -172,6 +172,9 @@ class HexagonalEdges(BaseEdgeBuilder):
def __init__(self, src_name: str, add_neighbouring_children: bool = False, depth_children: Optional[int] = 1):
super().__init__(src_name, src_name)
self.add_neighbouring_children = add_neighbouring_children

assert isinstance(depth_children, int), "Depth of children must be an integer"
assert depth_children > 0, "Depth of children must be positive"
self.depth_children = depth_children

def transform(self, graph: HeteroData, attrs_config: Optional[DotDict] = None) -> HeteroData:
Expand Down

0 comments on commit a59f5d1

Please sign in to comment.