post-run fix

amazon-braket · Oct 18, 2023 · 0a811a9 · 0a811a9
1 parent a2cfee1
commit 0a811a9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/..._Amazon_Braket_Hybrid_Jobs/Quantum_machine_learning_in_Amazon_Braket_Hybrid_Jobs_mocks.py b/..._Amazon_Braket_Hybrid_Jobs/Quantum_machine_learning_in_Amazon_Braket_Hybrid_Jobs_mocks.py
@@ -95,6 +95,6 @@ def post_run(tb):
         import os
         os.remove("model.tar.gz")
         os.remove("results.json")
+        mock_cloudpickle.stop()
         """
     )
-    mock_cloudpickle.stop()