Skip to content

Commit

Permalink
no reason to add dask dep just for this. hopefully the attr check aco…
Browse files Browse the repository at this point in the history
…mplishes what we want here.
  • Loading branch information
bnb32 committed Jul 24, 2024
1 parent 95d7b22 commit 840a487
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions rex/utilities/bc_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,6 @@
"""
import logging

import dask.array as da
import numpy as np
import scipy

Expand Down Expand Up @@ -193,7 +192,7 @@ def _clean_params(self, params, arr_shape):
"""

msg = f'params must be 2D array but received {type(params)}'
assert isinstance(params, (np.ndarray, da.core.Array)), msg
assert isinstance(hasattr(params, 'shape')), msg

if len(params.shape) == 1:
params = np.expand_dims(params, 0)
Expand Down

0 comments on commit 840a487

Please sign in to comment.