Better return type and better type comparison.

pasqal-io · Oct 24, 2024 · a3297eb · a3297eb
1 parent f97f542
commit a3297eb
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/qadence2_platforms/backends/pyqtorch/functions.py b/qadence2_platforms/backends/pyqtorch/functions.py
@@ -9,11 +9,11 @@
 from qadence2_platforms.backends.pyqtorch.utils import InputType, Support
 
 
-def _get_op(op: InputType) -> Primitive:
-    if op.is_symbol is True:  # needed to ensure the value (not the existence)
+def _get_op(op: InputType) -> Primitive | None:
+    if op.is_symbol is True:
         return getattr(pyq, op.head.upper(), None)
 
-    if op.is_quantum_operator is True:  # needed to ensure the value (not the existence)
+    if op.is_quantum_operator is True:
         op_args_item: InputType = op.args[0].args
         return getattr(pyq, op_args_item[0].upper(), None)