Skip to content

Commit fb82404

Browse files
committed
Move numba dependent import
1 parent 68a41ca commit fb82404

1 file changed

Lines changed: 3 additions & 1 deletion

File tree

scself/_denoise/n2s_denoise.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
import numpy as np
22
import scipy.sparse as sps
33

4-
from ..utils import dot, shrink_array_to_zero_inplace
4+
from scself.utils import dot
55
from .._noise2self.common import row_normalize
66
from .._noise2self.graph import combine_row_stochastic_graphs
77

@@ -207,6 +207,8 @@ def _denoise_chunk(
207207

208208
if zero_threshold is not None:
209209

210+
from scself.utils.shrink import shrink_array_to_zero_inplace
211+
210212
if sps.issparse(out):
211213
shrink_array_to_zero_inplace(
212214
out.data,

0 commit comments

Comments
 (0)