presumably less memory allocations

Author: bjarthur

src/TensorStoreWrapper.jl

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