File tree Expand file tree Collapse file tree 1 file changed +1
-9
lines changed Expand file tree Collapse file tree 1 file changed +1
-9
lines changed Original file line number Diff line number Diff line change @@ -572,7 +572,7 @@ def join_nonshared_inputs(
572
572
for var in vars :
573
573
shape = point [var .name ].shape
574
574
arr_len = np .prod (shape , dtype = int )
575
- replace [var ] = reshape_t ( inarray [last_idx : last_idx + arr_len ], shape ).astype (var .dtype )
575
+ replace [var ] = inarray [last_idx : last_idx + arr_len ]. reshape ( shape ).astype (var .dtype )
576
576
last_idx += arr_len
577
577
578
578
replace .update (shared )
@@ -581,14 +581,6 @@ def join_nonshared_inputs(
581
581
return xs_special , inarray
582
582
583
583
584
- def reshape_t (x , shape ):
585
- """Work around fact that x.reshape(()) doesn't work"""
586
- if shape != ():
587
- return x .reshape (shape )
588
- else :
589
- return x [0 ]
590
-
591
-
592
584
class PointFunc :
593
585
"""Wraps so a function so it takes a dict of arguments instead of arguments."""
594
586
You can’t perform that action at this time.
0 commit comments