Update src/TensorStoreWrapper.jl

Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>

Author: bjarthur

src/TensorStoreWrapper.jl

@@ -235,7 +235,8 @@ end
 Base.size(w::IndexDomainWrapper) = pyconvert(Tuple, parent(w).shape)
 function Base.size(w::IndexDomainWrapper, d::Integer)
     d < 1 && throw(ArgumentError("dimension must be ≥ 1"))
-    return d <= ndims(w) ? size(w)[d] : 1
+    di = Base.to_index(d)
+    return d <= ndims(w) ? size(w)[di] : 1
 end
 Base.ndims(w::IndexDomainWrapper) = pyconvert(Int, parent(w).rank)
 function Base.axes(w::IndexDomainWrapper)