diff --git a/data_tests/test_packaging/base.py b/data_tests/test_packaging/base.py index 8ff2207..b56f7dd 100644 --- a/data_tests/test_packaging/base.py +++ b/data_tests/test_packaging/base.py @@ -1,4 +1,5 @@ """A pure Python module containing a few functions""" + import numpy as np diff --git a/src/transonic/analyses/__init__.py b/src/transonic/analyses/__init__.py index a431841..975e73b 100644 --- a/src/transonic/analyses/__init__.py +++ b/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) diff --git a/src/transonic/analyses/util.py b/src/transonic/analyses/util.py index 16a52d7..0e3fce2 100644 --- a/src/transonic/analyses/util.py +++ b/src/transonic/analyses/util.py @@ -2,6 +2,7 @@ ============================= """ + import re from pathlib import Path from textwrap import dedent diff --git a/src/transonic/backends/cython.py b/src/transonic/backends/cython.py index bbb971d..21295a3 100644 --- a/src/transonic/backends/cython.py +++ b/src/transonic/backends/cython.py @@ -17,6 +17,7 @@ :private-members: """ + import copy import inspect diff --git a/src/transonic/backends/jax.py b/src/transonic/backends/jax.py index 4669b60..8caf29d 100644 --- a/src/transonic/backends/jax.py +++ b/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): diff --git a/src/transonic/config.py b/src/transonic/config.py index 032f8d4..b592c7f 100644 --- a/src/transonic/config.py +++ b/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). diff --git a/src/transonic/progress.py b/src/transonic/progress.py index e620a7c..a13ed20 100644 --- a/src/transonic/progress.py +++ b/src/transonic/progress.py @@ -5,6 +5,7 @@ logging messages. """ + from transonic.log import logger diff --git a/src/transonic/typing.py b/src/transonic/typing.py index 68934dd..128cd65 100644 --- a/src/transonic/typing.py +++ b/src/transonic/typing.py @@ -66,6 +66,7 @@ :private-members: """ + import re from enum import Enum, auto import itertools diff --git a/src/transonic_cl/init_meson.py b/src/transonic_cl/init_meson.py index 6a9cce7..aba456a 100644 --- a/src/transonic_cl/init_meson.py +++ b/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) -