Skip to content

Commit 3229c05

Browse files
Update src/neurostatslib/base_class.py
Co-authored-by: William F. Broderick <billbrod@gmail.com>
1 parent a4eb0eb commit 3229c05

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

src/neurostatslib/base_class.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -347,9 +347,8 @@ def _check_and_convert_params(
347347
raise ValueError("Params needs to be array-like of length two.")
348348

349349
try:
350-
params = jnp.asarray(params[0], dtype=data_type), jnp.asarray(
351-
params[1], dtype=data_type
352-
)
350+
params = (jnp.asarray(params[0], dtype=data_type),
351+
jnp.asarray(params[1], dtype=data_type))
353352
except (ValueError, TypeError):
354353
raise TypeError(
355354
"Initial parameters must be array-like of array-like objects"

0 commit comments

Comments
 (0)