Format with black

data_tests/test_packaging/base.py

@@ -1,4 +1,5 @@
 """A pure Python module containing a few functions"""
+
 import numpy as np
 
 

src/transonic/analyses/__init__.py

@@ -424,9 +424,9 @@ def analyse_aot(code, pathfile):
                 annotations["__locals__"][name_func] = annotations_locals
 
             if fdef.returns:
-                annotations["__returns__"][
-                    name_func
-                ] = extract_returns_annotation(fdef.returns, namespace)
+                annotations["__returns__"][name_func] = (
+                    extract_returns_annotation(fdef.returns, namespace)
+                )
 
     for signatures in annotations["__in_comments__"].values():
         replace_strings_by_objects(signatures, module, ancestors, udc, duc)

src/transonic/analyses/util.py

@@ -2,6 +2,7 @@
 =============================
 
 """
+
 import re
 from pathlib import Path
 from textwrap import dedent

src/transonic/backends/cython.py

@@ -17,6 +17,7 @@
    :private-members:
 
 """
+
 import copy
 import inspect
 

src/transonic/backends/jax.py

@@ -32,7 +32,8 @@ def add_jax_comments(code):
         # Replace `import numpy as np` -> `import jax.numpy as np`
         if isinstance(node, gast.Import):
             if (alias := node.names[0]).name == "numpy":
-                node = gast.Import([gast.alias(name="jax.numpy", asname=alias.asname or alias.name)])
+                g_alias = gast.alias(name="jax.numpy", asname=alias.asname or alias.name)
+                node = gast.Import([g_alias])
 
         # Replace `from numpy import eye` -> `from jax.numpy import eye`
         elif isinstance(node, gast.ImportFrom):

src/transonic/config.py

@@ -49,6 +49,7 @@
 path_root = Path(os.environ.get("TRANSONIC_DIR", Path.home() / ".transonic"))
 SUPPORTED_BACKENDS = frozenset(("pythran", "cython", "jax", "numba", "python"))
 
+
 def strtobool(value):
     """Convert a string representation of truth to true (1) or false (0).
 

src/transonic/progress.py

@@ -5,6 +5,7 @@
 logging messages.
 
 """
+
 from transonic.log import logger
 
 

src/transonic/typing.py

@@ -66,6 +66,7 @@
    :private-members:
 
 """
+
 import re
 from enum import Enum, auto
 import itertools

src/transonic_cl/init_meson.py

@@ -21,9 +21,9 @@ def process_directory(path_dir, path_pack=None):
     if path_pack is None:
         path_pack = path_dir
         print(f"Process {path_dir}")
-        subdir=path_pack.name
+        subdir = path_pack.name
     else:
-        subdir=PurePosixPath(path_dir.relative_to(path_pack.parent))
+        subdir = PurePosixPath(path_dir.relative_to(path_pack.parent))
         print(f"Process subdir {subdir}")
 
     names_py = sorted(path.name for path in path_dir.glob("*.py"))
@@ -77,4 +77,3 @@ def main():
         sys.exit(1)
 
     process_directory(path_pack)
-