one more mock

amazon-braket · Oct 12, 2023 · 5166690 · 5166690
1 parent 44597a3
commit 5166690
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/unit_tests/braket/jobs/test_hybrid_job.py b/test/unit_tests/braket/jobs/test_hybrid_job.py
@@ -420,10 +420,10 @@ def my_entry(*args):
     assert mock_tempdir.return_value.__exit__.called
 
 
-def test_serialization_error():
+def test_serialization_error(aws_session):
     ssl_context = SSLContext()
 
-    @hybrid_job(device=None)
+    @hybrid_job(device=None, aws_session=aws_session)
     def fails_serialization():
         print(ssl_context)