* Addressing review comment.

NVIDIA · Jul 21, 2024 · 27a689d · 27a689d
1 parent 228713f
commit 27a689d
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/python/cudaq/kernel/register_op.py b/python/cudaq/kernel/register_op.py
@@ -34,12 +34,10 @@ def kernel():
  if isinstance(unitary, Callable):
  raise RuntimeError("parameterized custom operations not yet supported.")
 
- if isinstance(unitary, np.matrix):
+ if isinstance(unitary, np.matrix) or isinstance(unitary, List):
  matrix = np.array(unitary)
  elif isinstance(unitary, np.ndarray):
  matrix = unitary
- elif isinstance(unitary, List):
- matrix = np.array(unitary)
  else:
  raise RuntimeError("unknown type of unitary.")