lp_notch_test.py: minor numpy usage cleanup

Avoids DeprecationWarning: Conversion of an array with ndim > 0 to a scalar starting with NumPy 1.25
BerkeleyLab · Dec 20, 2024 · 90903de · 90903de
1 parent 3111a33
commit 90903de
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dsp/hosted/lp_notch_test.py b/dsp/hosted/lp_notch_test.py
@@ -97,7 +97,7 @@ def __init__(
             Abp_zn = cl0_bp.response(zn)
             # solve for the final gains
             mat = numpy.array([[Alp_dc, Abp_dc], [Alp_zn, Abp_zn]])
-            target_gains = numpy.array([[self.gain], [self.ngain]])
+            target_gains = numpy.array([self.gain, self.ngain])
             raw_gains = numpy.linalg.inv(mat).dot(target_gains)
             # prevent the low-pass component from clipping
             if abs(raw_gains[0]) > 0.99: