Skip to content

Commit c18617b

Browse files
committed
use int dtype
1 parent cc8e70f commit c18617b

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

uxarray/grid/connectivity.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -473,7 +473,7 @@ def _build_node_edge_connectivity_numba(edge_nodes, n_node):
473473
n_edge, nodes_per_edge = edge_nodes.shape
474474

475475
# count how many edges touch each node
476-
counts = np.zeros(n_node, np.int64)
476+
counts = np.zeros(n_node, dtype=INT_DTYPE)
477477
for e in range(n_edge):
478478
for j in range(nodes_per_edge):
479479
node = edge_nodes[e, j]
@@ -487,9 +487,9 @@ def _build_node_edge_connectivity_numba(edge_nodes, n_node):
487487
max_edges = counts[i]
488488

489489
# allocate output, pad with fill
490-
node_edge = np.full((n_node, max_edges), INT_FILL_VALUE, dtype=np.int32)
490+
node_edge = np.full((n_node, max_edges), INT_FILL_VALUE, dtype=INT_DTYPE)
491491

492-
ptr = np.zeros(n_node, np.int64)
492+
ptr = np.zeros(n_node, dtype=INT_DTYPE)
493493

494494
# fill in
495495
for e in range(n_edge):

0 commit comments

Comments
 (0)