isort

heavyai · Jul 26, 2023 · 36b9b24 · 36b9b24
1 parent 447e2e3
commit 36b9b24
Show file tree

Hide file tree

Showing 11 changed files with 32 additions and 33 deletions.
diff --git 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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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