Fixes

tiledb/fragment.py

@@ -5,8 +5,6 @@
 import tiledb
 from tiledb.libtiledb import version as libtiledb_version
 
-from .main import PyFragmentInfo
-
 """
 Classes and functions relating to TileDB fragments.
 """
@@ -105,6 +103,8 @@ def __init__(self, array_uri, include_mbrs=False, ctx=None):
 
         self.array_uri = array_uri
 
+        from .main import PyFragmentInfo
+
         fi = PyFragmentInfo(self.array_uri, schema, include_mbrs, ctx)
 
         self.__nums = fi.get_num_fragments()

tiledb/query.py

@@ -3,7 +3,6 @@
 from typing import Optional, Sequence, Union
 
 import tiledb.libtiledb as lt
-from tiledb import TileDBError
 
 from .array import Array
 from .ctx import Ctx, CtxMixin, default_ctx
@@ -63,7 +62,7 @@ def __init__(
             raise ValueError("Cannot pass both dims and has_coords=True to Query")
 
         if return_incomplete and not array.schema.sparse:
-            raise TileDBError(
+            raise lt.TileDBError(
                 "Incomplete queries are only supported for sparse arrays at this time"
             )
 
@@ -82,15 +81,17 @@ def __init__(
             domain = array.schema.domain
             for dname in dims:
                 if not domain.has_dim(dname):
-                    raise TileDBError(f"Selected dimension does not exist: '{dname}'")
+                    raise lt.TileDBError(
+                        f"Selected dimension does not exist: '{dname}'"
+                    )
             self._dims = dims
         else:
             self._dims = None
 
         if attrs is not None:
             for name in attrs:
                 if not array.schema.has_attr(name):
-                    raise TileDBError(f"Selected attribute does not exist: '{name}'")
+                    raise lt.TileDBError(f"Selected attribute does not exist: '{name}'")
         self._attrs = attrs
         self._cond = cond
 
@@ -109,7 +110,9 @@ def __init__(
             if use_arrow is None:
                 use_arrow = True
             if not use_arrow:
-                raise TileDBError("Cannot initialize return_arrow with use_arrow=False")
+                raise lt.TileDBError(
+                    "Cannot initialize return_arrow with use_arrow=False"
+                )
         self._use_arrow = use_arrow
 
         self._return_incomplete = return_incomplete
@@ -124,7 +127,7 @@ def subarray(self) -> Subarray:
 
     def __getitem__(self, selection):
         if self._return_arrow:
-            raise TileDBError("`return_arrow=True` requires .df indexer`")
+            raise lt.TileDBError("`return_arrow=True` requires .df indexer`")
 
         return self._array.subarray(
             selection,