Linting

Author: kounelisagis

examples/global_order_writes.py

@@ -32,6 +32,7 @@
 #
 
 import numpy as np
+
 import tiledb
 
 # Name of the arrays to create
@@ -60,23 +61,23 @@ def create_dense_array():
 def write_sparse_global_order():
     """Write to sparse array in global order with multiple submits."""
     print("Writing sparse array in global order with multiple submits...")
-    
+
     with tiledb.open(sparse_array_name, "w") as A:
         # Create a query with global order layout
         q = tiledb.Query(A, order="G")
-        
+
         # First batch of data
         coords_batch1 = np.array([1, 5, 10, 20, 50], dtype=np.int32)
         data_batch1 = np.array([100, 200, 300, 400, 500], dtype=np.int64)
         q.set_data({"d1": coords_batch1, "a1": data_batch1})
         q.submit()
-        
+
         # Second batch of data (coordinates must be in global order relative to first batch)
         coords_batch2 = np.array([100, 200, 500], dtype=np.int32)
         data_batch2 = np.array([600, 700, 800], dtype=np.int64)
         q.set_data({"d1": coords_batch2, "a1": data_batch2})
         q.submit()
-        
+
         # Finalize to complete the write
         q.finalize()
 
@@ -89,29 +90,29 @@ def write_sparse_global_order():
 def write_dense_global_order():
     """Write to dense array in global order with multiple submits."""
     print("\nWriting dense array in global order with multiple submits...")
-    
+
     with tiledb.open(dense_array_name, "w") as A:
         # Create a query with global order layout
         q = tiledb.Query(A, order="G")
-        
+
         # Set the subarray to cover the full range we're writing
         start_coord = 1
         end_coord = 100
         q.set_subarray_ranges([(start_coord, end_coord)])
-        
+
         # First batch of data (cells 1-50)
         data_batch1 = np.arange(1, 51, dtype=np.int64)
         q.set_data({"a1": data_batch1})
         q.submit()
-        
+
         # Second batch of data (cells 51-100)
         data_batch2 = np.arange(51, 101, dtype=np.int64)
         q.set_data({"a1": data_batch2})
         q.submit()
-        
+
         # Finalize to complete the write
         q.finalize()
-    
+
     # Verify only one fragment was created
     fragments = tiledb.array_fragments(dense_array_name)
     print(f"Number of fragments created: {len(fragments)}")
@@ -121,13 +122,13 @@ def write_dense_global_order():
 def read_and_verify():
     """Read back the data to verify it was written correctly."""
     print("\nReading and verifying data...")
-    
+
     # Read sparse array
     with tiledb.open(sparse_array_name, "r") as A:
         result = A[:]
         print(f"Sparse array has {len(result['a1'])} cells")
         print(f"First 5 values: {result['a1'][:5]}")
-    
+
     # Read dense array
     with tiledb.open(dense_array_name, "r") as A:
         result = A[1:20]  # Read cells 1-19
@@ -138,20 +139,20 @@ def read_and_verify():
     print("=" * 60)
     print("TileDB Global Order Writes Example")
     print("=" * 60)
-    
+
     # Check if arrays exist, create if not
     if tiledb.object_type(sparse_array_name) != "array":
         create_sparse_array()
     if tiledb.object_type(dense_array_name) != "array":
         create_dense_array()
-    
+
     # Write data
     write_sparse_global_order()
     write_dense_global_order()
-    
+
     # Read and verify
     read_and_verify()
-    
+
     print("\n" + "=" * 60)
     print("Example completed successfully!")
     print("=" * 60)