fix: remove union from singledispatch

src/benchq/conversions/_operator_translations.py

@@ -7,7 +7,6 @@
     # Numpy throws deprecation warnings due to the scipy import
     from openfermion import QubitOperator, IsingOperator, InteractionOperator
 
-import openfermion
 from orquestra.integrations.cirq.conversions._openfermion_conversions import (
     to_openfermion,
 )
@@ -27,12 +26,22 @@ def get_pyliqtr_operator(hamiltonian):
 
 
 @get_pyliqtr_operator.register
-def _(hamiltonian: Union[PauliTerm, PauliSum]) -> Hamiltonian:
+def _(hamiltonian: PauliSum) -> Hamiltonian:
     return openfermion_to_pyliqtr(to_openfermion(hamiltonian))
 
 
 @get_pyliqtr_operator.register
-def _(hamiltonian: Union[QubitOperator, IsingOperator]) -> Hamiltonian:
+def _(hamiltonian: PauliTerm) -> Hamiltonian:
+    return openfermion_to_pyliqtr(to_openfermion(hamiltonian))
+
+
+@get_pyliqtr_operator.register
+def _(hamiltonian: QubitOperator) -> Hamiltonian:
+    return openfermion_to_pyliqtr(hamiltonian)
+
+
+@get_pyliqtr_operator.register
+def _(hamiltonian: IsingOperator) -> Hamiltonian:
     return openfermion_to_pyliqtr(hamiltonian)
 
 

tests/benchq/resource_estimation/graph_estimators/test_graph_estimator.py

@@ -10,7 +10,6 @@
     get_implementation_compiler,
     get_ruby_slippers_circuit_compiler,
 )
-from benchq.compilation.graph_states.compiled_data_structures import GSCInfo
 from benchq.decoder_modeling import DecoderModel
 from benchq.problem_embeddings.quantum_program import QuantumProgram
 from benchq.quantum_hardware_modeling import (