fix: circular imports

Author: sreevickrant

src/finchlite/__init__.py

@@ -1,55 +1,6 @@
-from .algebra import (
-    FTyped,
-    Tensor,
-    TensorFType,
-    bool,
-    bool_,
-    complex64,
-    complex128,
-    complex_,
-    ffuncs,
-    finfo,
-    fisinstance,
-    float16,
-    float32,
-    float64,
-    float_,
-    ftype,
-    iinfo,
-    int8,
-    int16,
-    int32,
-    int64,
-    int_,
-    intp,
-    isdtype,
-    none_,
-    str_,
-    uint8,
-    uint16,
-    uint32,
-    uint64,
-)
-from .autoschedule.tensor_stats import (
-    DC,
-    BlockedStats,
-    DCStats,
-    DenseStats,
-    TensorDef,
-    UniformStats,
-)
-from .codegen import (
-    NumpyBuffer,
-    NumpyBufferFType,
-)
-from .compile import (
-    AssemblyContext,
-    BufferizedNDArray,
-    Extent,
-    ExtentFType,
-    dimension,
-)
-from .finch_fused import jit
+# isort: split
+# interface must be imported first so EagerTensor is defined before the
+# autoschedule→compile chain runs (compile.BufferizedNDArray inherits from it).
 from .interface import (
     EagerTensor,
     LazyTensor,
@@ -161,6 +112,61 @@
     var,
     vecdot,
 )
+
+# isort: split
+
+from .algebra import (
+    FTyped,
+    Tensor,
+    TensorFType,
+    bool,
+    bool_,
+    complex64,
+    complex128,
+    complex_,
+    ffuncs,
+    finfo,
+    fisinstance,
+    float16,
+    float32,
+    float64,
+    float_,
+    ftype,
+    iinfo,
+    int8,
+    int16,
+    int32,
+    int64,
+    int_,
+    intp,
+    isdtype,
+    none_,
+    str_,
+    uint8,
+    uint16,
+    uint32,
+    uint64,
+)
+from .autoschedule.tensor_stats import (
+    DC,
+    BlockedStats,
+    DCStats,
+    DenseStats,
+    TensorDef,
+    UniformStats,
+)
+from .codegen import (
+    NumpyBuffer,
+    NumpyBufferFType,
+)
+from .compile import (
+    AssemblyContext,
+    BufferizedNDArray,
+    Extent,
+    ExtentFType,
+    dimension,
+)
+from .finch_fused import jit
 from .tensor import (
     DenseLevel,
     DenseLevelFType,

src/finchlite/compile/bufferized_ndarray.py

@@ -8,6 +8,7 @@
 from ..algebra import FType, ImmutableStructFType, Tensor, TupleFType, ffuncs, ftype
 from ..codegen import NumpyBuffer, NumpyBufferFType
 from ..codegen.numba_codegen import to_numpy_type
+from ..interface.eager import EagerTensor
 from . import looplets as lplt
 from .lower import AssemblyContext, FinchTensorFType
 
@@ -16,7 +17,7 @@ def _get_default_strides(size: tuple[int, ...]) -> tuple[int, ...]:
     return tuple(np.cumprod((1,) + size[::-1]).astype(int))[-2::-1]
 
 
-class BufferizedNDArray(Tensor):
+class BufferizedNDArray(EagerTensor):
     def __init__(
         self,
         val: NumpyBuffer,

src/finchlite/interface/eager.py

@@ -6,8 +6,6 @@
 from typing import Any
 
 from ..algebra import FinchOperator
-from . import lazy
-from .fuse import compute
 from .overrides import OverrideTensor
 
 
@@ -231,6 +229,10 @@ def __ne__(self, other):
         return not_equal(self, other)
 
 
+from . import lazy  # noqa: E402
+from .fuse import compute  # noqa: E402
+
+
 def full(
     shape: int | tuple[int, ...],
     fill_value: bool | complex,
@@ -418,6 +420,7 @@ def matmul(x1, x2, /):
     """
     if isinstance(x1, lazy.LazyTensor) or isinstance(x2, lazy.LazyTensor):
         return lazy.matmul(x1, x2)
+
     c = lazy.matmul(x1, x2)
     return compute(c)