Fix global tqdm module patch in test (#2319)

unitaryfund · Apr 29, 2024 · b7b6e1e · b7b6e1e
1 parent a134e52
commit b7b6e1e
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/mitiq/shadows/tests/test_quantum_processing.py b/mitiq/shadows/tests/test_quantum_processing.py
@@ -45,6 +45,10 @@ def test_tqdm_import_not_available():
         )  # Reload the module to trigger the import
         assert mitiq.shadows.quantum_processing.tqdm is None
 
+    # Reload the module again to restore the original tqdm import.
+    # Otherwise, the rest of the tests are affected by the patch (issue #2318)
+    importlib.reload(mitiq.shadows.quantum_processing)
+
 
 def test_generate_random_pauli_strings():
     """Tests that the function generates random Pauli strings."""