make format

PennyLaneAI · Dec 17, 2024 · 511ea57 · 511ea57
1 parent 8bb0129
commit 511ea57
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/frontend/catalyst/third_party/cuda/catalyst_to_cuda_interpreter.py b/frontend/catalyst/third_party/cuda/catalyst_to_cuda_interpreter.py
@@ -432,7 +432,13 @@ def change_instruction(ctx, eqn):
 
     # Now, we can map to the correct op
     # For now just assume rx
-    cuda_inst(ctx.kernel, *qubits_or_params, inst=cuda_inst_name, qubits_len=qubits_len, static_params=static_params)
+    cuda_inst(
+        ctx.kernel,
+        *qubits_or_params,
+        inst=cuda_inst_name,
+        qubits_len=qubits_len,
+        static_params=static_params,
+    )
 
     # Finally determine how many are qubits.
     qubits = qubits_or_params[:qubits_len]

diff --git a/frontend/catalyst/third_party/cuda/primitives/__init__.py b/frontend/catalyst/third_party/cuda/primitives/__init__.py
@@ -307,7 +307,9 @@ def gate_func(kernel, *qubits_or_params, inst=None, qubits_len=-1, static_params
         Quantum operations in CUDA-quantum return no values. But JAXPR expects return values.
         We can just say that multiple_results = True and return an empty tuple.
         """
-        kernel_gate_p.bind(kernel, *qubits_or_params, inst=inst, qubits_len=qubits_len, static_params=static_params)
+        kernel_gate_p.bind(
+            kernel, *qubits_or_params, inst=inst, qubits_len=qubits_len, static_params=static_params
+        )
         return tuple()
 
     @kernel_gate_p.def_impl