codefactor

frontend/catalyst/jit.py

@@ -38,7 +38,6 @@
 from catalyst.jax_tracer import lower_jaxpr_to_mlir, trace_to_jaxpr
 from catalyst.logging import debug_logger, debug_logger_init
 from catalyst.passes import _inject_transform_named_sequence
-from catalyst.passes import pipeline as circuit_transform_pass_pipeline
 from catalyst.qfunc import QFunc
 from catalyst.tracing.contexts import EvaluationContext
 from catalyst.tracing.type_signatures import (

frontend/catalyst/passes.py

@@ -79,7 +79,7 @@ def wrapper(*args, **kwrags):
         return wrapped_qnode_function(*args, **kwrags)
 
     fn_clone.func = wrapper
-    fn_clone._peephole_transformed = True
+    fn_clone._peephole_transformed = True # pylint: disable=protected-access
 
     return fn_clone
 
@@ -209,6 +209,7 @@ def wrapper(*args, **kwrags):
 
 
 ## IMPL and helpers ##
+# pylint: disable=missing-function-docstring
 class _PipelineNameUniquer:
     def __init__(self, i):
         self.i = i

frontend/catalyst/qfunc.py

@@ -109,6 +109,7 @@ def extract_backend_info(
         return extract_backend_info(device, capabilities)
 
     # pylint: disable=no-member
+    # pylint: disable=self-cls-assignment
     @debug_logger
     def __call__(self, *args, **kwargs):
         assert isinstance(self, qml.QNode)
@@ -119,7 +120,7 @@ def __call__(self, *args, **kwargs):
             if not hasattr(self, "_peephole_transformed"):
                 self = pipeline(pass_pipeline=pass_pipeline)(
                     self
-                )  # pylint: disable=self-cls-assignment
+                )
             kwargs.pop("pass_pipeline")
 
         # Mid-circuit measurement configuration/execution

frontend/test/lit/test_peephole_optimizations.py

@@ -148,7 +148,7 @@ def f(x):
 
     @qjit(keep_intermediate=True)
     def test_pipeline_lowering_keep_original_workflow(x):
-        return f(1.2), f_pipeline(1.2)
+        return f(x), f_pipeline(x)
 
     # CHECK: transform_named_sequence
     # CHECK: call_jaxpr=