Remove old argument in WASP pycuda because of #520 (#533)

ptycho · Feb 2, 2024 · 1798bd1 · 1798bd1
1 parent a0a4b7e
commit 1798bd1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ptypy/custom/WASP_pycuda.py b/ptypy/custom/WASP_pycuda.py
@@ -69,9 +69,9 @@ def engine_initialize(self):
         Prepare for reconstruction.
         """
         # Context, Multi GPU communicator and Stream (needs to be in this order)
-        self.context, self.queue = get_context(new_context=True, new_queue=False)
+        self.context, self.queue = get_context(new_queue=False)
         self.multigpu = get_multi_gpu_communicator()
-        self.context, self.queue = get_context(new_context=False, new_queue=True)
+        self.context, self.queue = get_context(new_queue=True)
 
         # initialise kernels for centring probe if required
         if self.p.probe_center_tol is not None: