Format files

Author: kounelisagis

examples/variable_length.py

@@ -35,13 +35,13 @@
 
 import numpy as np
 
-#%%
+# %%
 import tiledb
 from tiledb.tests.common import assert_subarrays_equal
 
 array_name = "variable_length_array"
 
-#%%
+# %%
 
 
 def create_array():

tiledb/__init__.py

@@ -108,6 +108,7 @@
 __version__ = version.version
 group_create = Group.create
 
+
 # Create a proxy object to wrap libtiledb and provide a `cc` alias
 class CCProxy:
     def __init__(self, module):

tiledb/dataframe_.py

@@ -692,9 +692,9 @@ def _create_array(uri, df, sparse, full_domain, index_dims, column_infos, tiledb
             enums=enums,
             cell_order=tiledb_args["cell_order"],
             tile_order=tiledb_args["tile_order"],
-            coords_filters=None
-            if coord_filter is True
-            else _get_schema_filters(coord_filter),
+            coords_filters=(
+                None if coord_filter is True else _get_schema_filters(coord_filter)
+            ),
             offsets_filters=_get_schema_filters(
                 tiledb_args.get("offsets_filters", True)
             ),

tiledb/dense_array.py

@@ -281,10 +281,10 @@ def subarray(self, selection, attrs=None, cond=None, coords=False, order=None):
         out = self._read_dense_subarray(subarray, attr_names, cond, layout, coords)
         if any(s.step for s in idx):
             steps = tuple(slice(None, None, s.step) for s in idx)
-            for (k, v) in out.items():
+            for k, v in out.items():
                 out[k] = v.__getitem__(steps)
         if drop_axes:
-            for (k, v) in out.items():
+            for k, v in out.items():
                 out[k] = v.squeeze(axis=drop_axes)
         # attribute is anonymous, just return the result
         if not coords and self.schema.nattr == 1:

tiledb/domain.py

@@ -147,7 +147,7 @@ def size(self):
         """
         if not np.issubdtype(self.dtype, self.integer):
             raise TypeError("size valid only for integer domains")
-        return np.product(self.shape)
+        return np.prod(self.shape)
 
     def _is_homogeneous(self):
         dtype0 = self.dim(0).dtype

tiledb/filter.py

@@ -879,12 +879,10 @@ def chunksize(self):
         return self._chunksize
 
     @overload
-    def __getitem__(self, idx: int) -> Filter:
-        ...
+    def __getitem__(self, idx: int) -> Filter: ...
 
     @overload
-    def __getitem__(self, idx: slice) -> List[Filter]:
-        ...
+    def __getitem__(self, idx: slice) -> List[Filter]: ...
 
     def __getitem__(self, idx):
         """Gets a copy of the filter in the list at the given index

tiledb/tests/test_core.py

@@ -57,10 +57,13 @@ def test_pyquery_init(self):
                 self.assertEqual(q._test_init_buffer_bytes, intmax)
                 self.assertEqual(q._test_alloc_max_bytes, intmax)
 
-                with self.assertRaisesRegex(
-                    ValueError,
-                    "Invalid parameter: 'py.alloc_max_bytes' must be >= 1 MB ",
-                ), tiledb.scope_ctx({"py.alloc_max_bytes": 10}) as ctx2:
+                with (
+                    self.assertRaisesRegex(
+                        ValueError,
+                        "Invalid parameter: 'py.alloc_max_bytes' must be >= 1 MB ",
+                    ),
+                    tiledb.scope_ctx({"py.alloc_max_bytes": 10}) as ctx2,
+                ):
                     q = core.PyQuery(ctx2, a, ("",), (), 0, False)
 
     def test_import_buffer(self):

tiledb/tests/test_libtiledb.py

@@ -2832,9 +2832,10 @@ def test_pickle_with_tuple_timestamps(self, sparse, use_timestamps):
                         assert_array_equal(T[:], T2[:])
                     assert T2.timestamp_range == (timestamps[1], timestamps[2])
 
-            with io.BytesIO() as buf, tiledb.open(
-                path, timestamp=(timestamps[1], timestamps[2])
-            ) as V:
+            with (
+                io.BytesIO() as buf,
+                tiledb.open(path, timestamp=(timestamps[1], timestamps[2])) as V,
+            ):
                 pickle.dump(V, buf)
                 buf.seek(0)
                 with pickle.load(buf) as V2:

tiledb/tests/test_multi_index.py

@@ -436,9 +436,7 @@ def test_multirange_1d_sparse_double(self):
             assert_array_equal(orig_array[-1], A.multi_index[30.0][attr_name])
             assert_array_equal(
                 orig_array[coords.size - 3 : coords.size],
-                A.multi_index[
-                    (28.0, 30.0),
-                ][attr_name],
+                A.multi_index[(28.0, 30.0),][attr_name],
             )
 
             res = A.multi_index[slice(0, 5)]