diff --git a/fdfd_tools/vectorization.py b/fdfd_tools/vectorization.py index bdc9d6a..57b58fb 100644 --- a/fdfd_tools/vectorization.py +++ b/fdfd_tools/vectorization.py @@ -45,5 +45,5 @@ def unvec(v: vfield_t, shape: numpy.ndarray) -> field_t: """ if numpy.any(numpy.equal(v, None)): return None - return vi.reshape((3, *shape), order='C') + return v.reshape((3, *shape), order='C')