From 36b9b24441543e9e617655d5100ee5956bd56ebb Mon Sep 17 00:00:00 2001 From: Guilherme Leobas Date: Wed, 26 Jul 2023 20:25:44 +0000 Subject: [PATCH] isort --- rbc/stdlib/__init__.py | 8 +++++--- rbc/stdlib/constants.py | 1 - rbc/stdlib/creation_functions.py | 9 +++++---- rbc/stdlib/data_type_functions.py | 6 ++++-- rbc/stdlib/elementwise_functions.py | 13 ++++--------- rbc/stdlib/manipulation_functions.py | 7 ++++--- rbc/stdlib/searching_functions.py | 2 +- rbc/stdlib/set_functions.py | 5 +++-- rbc/stdlib/statistical_functions.py | 10 +++++----- rbc/stdlib/utility_functions.py | 2 +- rbc/tests/heavydb/test_array_api_unsupported.py | 2 -- 11 files changed, 32 insertions(+), 33 deletions(-) diff --git a/rbc/stdlib/__init__.py b/rbc/stdlib/__init__.py index 7735bb9d2..0cac04052 100644 --- a/rbc/stdlib/__init__.py +++ b/rbc/stdlib/__init__.py @@ -1,11 +1,13 @@ -import textwrap import functools -import numpy as np +import textwrap from enum import Enum + +import numpy as np from numba.core import extending from numba.np.numpy_support import as_dtype, from_dtype + +from rbc import errors, typesystem from rbc.heavydb import Array, ArrayPointer -from rbc import typesystem, errors ARRAY_API_ADDRESS = ("https://data-apis.org/array-api/latest/API_specification/" "generated/array_api.{0}.html#array_api.{0}") diff --git a/rbc/stdlib/constants.py b/rbc/stdlib/constants.py index 3b99ff915..0d6a3bd10 100644 --- a/rbc/stdlib/constants.py +++ b/rbc/stdlib/constants.py @@ -5,7 +5,6 @@ """ import numpy as np - __all__ = [ 'e', 'inf', 'nan', 'pi' ] diff --git a/rbc/stdlib/creation_functions.py b/rbc/stdlib/creation_functions.py index 1e8f6a82a..53ddc1be9 100644 --- a/rbc/stdlib/creation_functions.py +++ b/rbc/stdlib/creation_functions.py @@ -4,13 +4,14 @@ https://data-apis.org/array-api/latest/API_specification/creation_functions.html """ -from rbc import typesystem, errors -from rbc.heavydb import Array, ArrayPointer -from rbc.stdlib import Expose, API -from numba.core import extending, types, cgutils from numba import TypingError +from numba.core import cgutils, extending, types from numba.core.typing import asnumbatype +from rbc import errors, typesystem +from rbc.heavydb import Array, ArrayPointer +from rbc.stdlib import API, Expose + __all__ = [ 'full', 'full_like', 'empty_like', 'empty', 'zeros', 'zeros_like', 'ones', 'ones_like', 'cumsum', 'arange', 'asarray', diff --git a/rbc/stdlib/data_type_functions.py b/rbc/stdlib/data_type_functions.py index e54accf65..f77cbe310 100644 --- a/rbc/stdlib/data_type_functions.py +++ b/rbc/stdlib/data_type_functions.py @@ -3,10 +3,12 @@ https://data-apis.org/array-api/latest/API_specification/data_type_functions.html """ -import numpy as np from typing import NamedTuple -from numba.core import types, extending + +import numpy as np +from numba.core import extending, types from numba.np.numpy_support import as_dtype, from_dtype + from rbc.errors import NumbaTypeError from rbc.heavydb import ArrayPointer from rbc.stdlib import Expose diff --git a/rbc/stdlib/elementwise_functions.py b/rbc/stdlib/elementwise_functions.py index a8b805d5a..bb0179155 100644 --- a/rbc/stdlib/elementwise_functions.py +++ b/rbc/stdlib/elementwise_functions.py @@ -4,18 +4,13 @@ https://data-apis.org/array-api/latest/API_specification/elementwise_functions.html. """ -from rbc.stdlib import ( - Expose, - BinaryUfuncExpose, - UnaryUfuncExpose, - API, - determine_input_type, -) import numpy as np -from rbc import typesystem -from rbc.heavydb import ArrayPointer, Array from numba.core import types +from rbc import typesystem +from rbc.heavydb import Array, ArrayPointer +from rbc.stdlib import (API, BinaryUfuncExpose, Expose, UnaryUfuncExpose, + determine_input_type) __all__ = [ 'add', 'subtract', 'multiply', 'divide', 'logaddexp', 'logaddexp2', diff --git a/rbc/stdlib/manipulation_functions.py b/rbc/stdlib/manipulation_functions.py index b7151474d..b08ce41b2 100644 --- a/rbc/stdlib/manipulation_functions.py +++ b/rbc/stdlib/manipulation_functions.py @@ -3,11 +3,12 @@ https://data-apis.org/array-api/latest/API_specification/manipulation_functions.html """ -from rbc.stdlib import Expose -from rbc.heavydb import ArrayPointer -from rbc.errors import NumbaTypeError from numba import literal_unroll +from rbc.errors import NumbaTypeError +from rbc.heavydb import ArrayPointer +from rbc.stdlib import Expose + __all__ = [ "concat", "expand_dims", diff --git a/rbc/stdlib/searching_functions.py b/rbc/stdlib/searching_functions.py index bdaf2bcca..52aa2f489 100644 --- a/rbc/stdlib/searching_functions.py +++ b/rbc/stdlib/searching_functions.py @@ -3,8 +3,8 @@ https://data-apis.org/array-api/latest/API_specification/searching_functions.html """ -from rbc.stdlib import Expose from rbc.heavydb import Array, ArrayPointer, type_can_asarray +from rbc.stdlib import Expose __all__ = ["argmax", "argmin", "nonzero", "where"] diff --git a/rbc/stdlib/set_functions.py b/rbc/stdlib/set_functions.py index 9c03aafa5..a106ac638 100644 --- a/rbc/stdlib/set_functions.py +++ b/rbc/stdlib/set_functions.py @@ -3,10 +3,11 @@ https://data-apis.org/array-api/latest/API_specification/set_functions.html """ -from rbc.stdlib import Expose -from rbc.heavydb import Array, ArrayPointer from numba.core import types +from rbc.heavydb import Array, ArrayPointer +from rbc.stdlib import Expose + __all__ = ["unique_all", "unique_counts", "unique_inverse", "unique_values"] expose = Expose(globals(), "set_functions") diff --git a/rbc/stdlib/statistical_functions.py b/rbc/stdlib/statistical_functions.py index 427060a64..c04ce3da5 100644 --- a/rbc/stdlib/statistical_functions.py +++ b/rbc/stdlib/statistical_functions.py @@ -3,14 +3,14 @@ https://data-apis.org/array-api/latest/API_specification/statistical_functions.html """ -from rbc.externals.stdio import printf +import numpy as np +from numba.core import errors, extending, types +from numba.np import numpy_support + from rbc import typesystem +from rbc.externals.stdio import printf from rbc.heavydb import ArrayPointer from rbc.stdlib import Expose -from numba.core import extending, types, errors -from numba.np import numpy_support -import numpy as np - __all__ = [ 'min', 'max', 'mean', 'prod', 'sum', 'std', 'var' diff --git a/rbc/stdlib/utility_functions.py b/rbc/stdlib/utility_functions.py index 019ca5d4e..eec1de5b6 100644 --- a/rbc/stdlib/utility_functions.py +++ b/rbc/stdlib/utility_functions.py @@ -3,8 +3,8 @@ https://data-apis.org/array-api/latest/API_specification/utility_functions.html """ -from rbc.stdlib import Expose from rbc.heavydb import ArrayPointer +from rbc.stdlib import Expose __all__ = ["all", "any"] diff --git a/rbc/tests/heavydb/test_array_api_unsupported.py b/rbc/tests/heavydb/test_array_api_unsupported.py index 57bd46126..3efc731cc 100644 --- a/rbc/tests/heavydb/test_array_api_unsupported.py +++ b/rbc/tests/heavydb/test_array_api_unsupported.py @@ -17,8 +17,6 @@ def heavydb(): "expand_dims", "flip", "permute_dims", "reshape", "roll", "squeeze", "stack", # sorting_functions "argsort", "sort", - # searching_functions - "where", # elementwise_functions 'float_power', 'divmod', 'cbrt', 'isnat', # set_functions