diff --git a/mmwave/dsp/angle_estimation.py b/mmwave/dsp/angle_estimation.py index 19e0d87f..d955aae1 100644 --- a/mmwave/dsp/angle_estimation.py +++ b/mmwave/dsp/angle_estimation.py @@ -526,7 +526,7 @@ def gen_steering_vec(ang_est_range, ang_est_resolution, num_ant): real = np.cos(mag) imag = np.sin(mag) - steering_vectors[kk, jj] = np.complex(real, imag) + steering_vectors[kk, jj] = np.complex64(real + 1j*imag) return [num_vec, steering_vectors]