Skip to content

Commit ae83e9b

Browse files
committed
Allow float input to BaseCircuit.predict function
1 parent 2407fc1 commit ae83e9b

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

impedance/models/circuits/circuits.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -133,7 +133,10 @@ def predict(self, frequencies, use_initial=False):
133133
impedance: ndarray of dtype 'complex128'
134134
Predicted impedance at each frequency
135135
"""
136-
frequencies = np.array(frequencies, dtype=float)
136+
frequencies = (
137+
np.array([frequencies], dtype=float) if isinstance(frequencies, float)
138+
else np.array(frequencies, dtype=float)
139+
)
137140

138141
if self._is_fit() and not use_initial:
139142
return eval(buildCircuit(self.circuit, frequencies,

0 commit comments

Comments
 (0)