formatting

Signed-off-by: Sachin Pisal <[email protected]>
NVIDIA · Nov 22, 2024 · a790d8a · a790d8a
1 parent 0d0cd14
commit a790d8a
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 8 deletions.
diff --git a/python/cudaq/operator/cudm_helpers.py b/python/cudaq/operator/cudm_helpers.py
@@ -134,8 +134,8 @@ def mul(
                                                   coeff=coeff1 * coeff2)
         return new_opterm
 
-    def _scalar_to_op(
-            self, scalar: CallbackCoefficient | Number) -> CudmOperatorTerm:
+    def _scalar_to_op(self,
+                      scalar: CallbackCoefficient | Number) -> CudmOperatorTerm:
         op_mat = numpy.identity(self._dimensions[0], dtype=numpy.complex128)
         op_term = cudm.tensor_product((cudm.DenseOperator(op_mat), (0,)),
                                       coeff=scalar)

diff --git a/python/cudaq/operator/integrators/builtin_integrators.py b/python/cudaq/operator/integrators/builtin_integrators.py
@@ -17,12 +17,12 @@
 except ImportError:
     has_cupy = False
 
+
 class cuDensityMatTimeStepper(BaseTimeStepper[CudmStateType]):
 
     def __init__(self, liouvillian: CudmOperator, ctx: CudmWorkStream):
         if not has_cupy:
-            raise ImportError(
-                'CuPy is required to use integrators.')
+            raise ImportError('CuPy is required to use integrators.')
         self.liouvillian = liouvillian
         self.ctx = ctx
         self.state = None
@@ -58,8 +58,7 @@ def __init__(self,
                  stepper: BaseTimeStepper[CudmStateType] = None,
                  **kwargs):
         if not has_cupy:
-            raise ImportError(
-                'CuPy is required to use integrators.')
+            raise ImportError('CuPy is required to use integrators.')
         super().__init__(**kwargs)
         self.stepper = stepper
 

diff --git a/python/cudaq/operator/integrators/scipy_integrators.py b/python/cudaq/operator/integrators/scipy_integrators.py
@@ -24,6 +24,7 @@
 except ImportError:
     has_cupy = False
 
+
 class ScipyZvodeIntegrator(BaseIntegrator[CudmStateType]):
     n_steps = 2500
     atol = 1e-8
@@ -34,8 +35,7 @@ def __init__(self, stepper: BaseTimeStepper[CudmStateType], **kwargs):
         if not has_scipy:
             raise ImportError("scipy is required to use this integrator.")
         if not has_cupy:
-            raise ImportError(
-                'CuPy is required to use this integrator.')
+            raise ImportError('CuPy is required to use this integrator.')
         super().__init__(**kwargs)
         self.stepper = stepper
         self.state_data_shape = None