Fix typing issues

pasqal-io · Nov 19, 2024 · 67197da · 67197da
1 parent 09e42e7
commit 67197da
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/pulser-core/pulser/backend/state.py b/pulser-core/pulser/backend/state.py
@@ -17,11 +17,11 @@
 from abc import ABC, abstractmethod
 from collections import Counter
 from collections.abc import Sequence
-from typing import Generic, Literal, Type, TypeVar
+from typing import Generic, Literal, Type, TypeVar, Union
 
 from pulser.channels.base_channel import States
 
-Eigenstate = States | Literal["O", "1"]
+Eigenstate = Union[States, Literal["0", "1"]]
 
 ArgScalarType = TypeVar("ArgScalarType")
 ReturnScalarType = TypeVar("ReturnScalarType")

diff --git a/pulser-pasqal/pulser_pasqal/pasqal_cloud.py b/pulser-pasqal/pulser_pasqal/pasqal_cloud.py
@@ -238,6 +238,7 @@ def _query_job_progress(
         get_batch_fn = backoff_decorator(self._sdk_connection.get_batch)
         batch = get_batch_fn(id=batch_id)
 
+        assert isinstance(batch.sequence_builder, str)
         seq_builder = Sequence.from_abstract_repr(batch.sequence_builder)
         reg = seq_builder.get_register(include_mappable=True)
         all_qubit_ids = reg.qubit_ids