formatting

Signed-off-by: Bettina Heim <[email protected]>

Author: bettinaheim

python/cudaq/kernel/analysis.py

@@ -56,13 +56,15 @@ def visit_FunctionDef(self, node):
         [self.visit(stm) for stm in node.body]
 
     def visit_Attribute(self, node):
-        if not self.kernelName: return
+        if not self.kernelName:
+            return
         if node.attr in globalAstRegistry:
             self.depKernels[node.attr] = globalAstRegistry[node.attr]
         self.visit(node.value)
 
     def visit_Name(self, node):
-        if not self.kernelName: return
+        if not self.kernelName:
+            return
         if node.id in globalAstRegistry:
             self.depKernels[node.id] = globalAstRegistry[node.id]
 

python/cudaq/kernel/ast_bridge.py

@@ -187,22 +187,24 @@ def emitFatalErrorOverride(msg):
                     f"Callable type must have signature specified ({ast.unparse(annotation) if hasattr(ast, 'unparse') else annotation})."
                 )
 
-            if hasattr(annotation.slice, 'elts') and len(annotation.slice.elts) == 2:
+            if hasattr(annotation.slice, 'elts') and len(
+                    annotation.slice.elts) == 2:
                 args = annotation.slice.elts[0]
                 ret = annotation.slice.elts[1]
             elif hasattr(annotation.slice, 'value') and hasattr(
-                    annotation.slice.value, 'elts') and len(annotation.slice.value.elts) == 2:
+                    annotation.slice.value, 'elts') and len(
+                        annotation.slice.value.elts) == 2:
                 args = annotation.slice.value.elts[0]
                 ret = annotation.slice.value.elts[1]
             else:
                 localEmitFatalError(
                     f"Unable to get list elements when inferring type from annotation ({ast.unparse(annotation) if hasattr(ast, 'unparse') else annotation})."
                 )
-            argTypes = [
-                mlirTypeFromAnnotation(a, ctx) for a in args.elts
-            ]
+            argTypes = [mlirTypeFromAnnotation(a, ctx) for a in args.elts]
             if not isinstance(ret, ast.Constant) or ret.value:
-                localEmitFatalError("passing kernels as arguments that return a value is not currently supported")
+                localEmitFatalError(
+                    "passing kernels as arguments that return a value is not currently supported"
+                )
             return cc.CallableType.get(argTypes)
 
         if isinstance(annotation,