cleanup

PennyLaneAI · Nov 14, 2024 · 90ba867 · 90ba867
1 parent 89fe923
commit 90ba867
Showing 1 changed file with 6 additions and 8 deletions.
diff --git a/pennylane_lightning/core/_serialize.py b/pennylane_lightning/core/_serialize.py
@@ -451,15 +451,13 @@ def get_wires(operation, single_op):
                 if isinstance(single_op, qml.QubitUnitary):
                     params.append([0.0])
                     mats.append(matrix(single_op))
-                elif inverse and not hasattr(self.sv_type, single_op.base.name):
-                    params.append([])
-                    mats.append(matrix(single_op))
-                elif not inverse and not hasattr(self.sv_type, name):
-                    params.append([])
-                    mats.append(matrix(single_op))
                 else:
-                    params.append(single_op.parameters)
-                    mats.append([])
+                    if hasattr(self.sv_type, single_op.base.name if inverse else name):
+                        params.append(single_op.parameters)
+                        mats.append([])
+                    else:
+                        params.append([])
+                        mats.append(matrix(single_op))
 
                 controlled_values.append(controlled_values_list)
                 controlled_wires.append(