Skip to content

Commit

Permalink
fix(csr): Set type of node_props to float64
Browse files Browse the repository at this point in the history
`dtype` of `node_props` passed from `csr_to_nbgraph()` to `NBGraph` was `float32`. This sets the type to explicitly be
`np.float64` which matches the expectation of `csr_spec_float` passed to `numba.experimental.jitclass()`.
  • Loading branch information
ns-rse committed Oct 30, 2024
1 parent 5184bfa commit 17e7661
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/skan/csr.py
Original file line number Diff line number Diff line change
Expand Up @@ -207,7 +207,7 @@ def csr_to_nbgraph(csr, node_props=None):
csr.indices,
csr.data,
np.array(csr.shape, dtype=np.int32),
node_props,
node_props.astype(np.float64),
)


Expand Down

0 comments on commit 17e7661

Please sign in to comment.