Resolving the type ignore issue

BQSKit · Nov 8, 2024 · b49b729 · b49b729
1 parent a66f777
commit b49b729
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/qfactorjax/qfactor_sample_jax.py b/qfactorjax/qfactor_sample_jax.py
@@ -389,11 +389,10 @@ def safe_call_jited_vmaped_state_sample_sweep(
                 training_states_kets,
             )
 
-            # TODO: Fix the typing ignore here
             results = tuple(
                 jnp.concatenate((results1[i], results2[i]))
                 for i in range(5)
-            )  # type: ignore
+            )
 
         return results