Fix value error bug with validating mlirTypeFromPyType for np.ndarray (…

…#1406)

* Fix value error bug with validating mlirTypeFromPyType for np.array

Signed-off-by: Alex McCaskey <amccaskey@nvidia.com>

* add the test

Signed-off-by: Alex McCaskey <amccaskey@nvidia.com>

* fix ci failure

Signed-off-by: Alex McCaskey <amccaskey@nvidia.com>

* Move new test due to issue #1400

---------

Signed-off-by: Alex McCaskey <amccaskey@nvidia.com>
Co-authored-by: Ben Howe <bhowe@nvidia.com>

python/cudaq/kernel/utils.py

@@ -173,8 +173,11 @@ def mlirTypeFromPyType(argType, ctx, **kwargs):
         return ComplexType.get(mlirTypeFromPyType(float, ctx))
 
     if argType in [list, np.ndarray, List]:
-        if 'argInstance' not in kwargs or kwargs['argInstance'] == None:
+        if 'argInstance' not in kwargs:
             return cc.StdvecType.get(ctx, mlirTypeFromPyType(float, ctx))
+        if argType != np.ndarray:
+            if kwargs['argInstance'] == None:
+                return cc.StdvecType.get(ctx, mlirTypeFromPyType(float, ctx))
 
         argInstance = kwargs['argInstance']
         argTypeToCompareTo = kwargs[

python/tests/kernel/test_kernel_features.py

@@ -851,6 +851,17 @@ def test2() -> int:
     assert test2() == 10
 
 
+def test_no_valueerror_np_array():
+
+    @cudaq.kernel
+    def test(var: np.ndarray):
+        q = cudaq.qubit()
+        ry(var[0], q)
+        mz(q)
+
+    test(np.array([1., 2.]))
+
+
 def test_draw():
 
     @cudaq.kernel