import Base functions

stecrotti · Mar 28, 2024 · 2029bde · 2029bde
1 parent 60c797f
commit 2029bde
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/src/periodic_tensor_train.jl b/src/periodic_tensor_train.jl
@@ -19,8 +19,9 @@ function PeriodicTensorTrain(tensors::Vector{Array{F,N}}) where {F<:Number, N}
     return PeriodicTensorTrain{F,N}(tensors)
 end
 
-@forward PeriodicTensorTrain.tensors getindex, iterate, firstindex, lastindex, setindex!, 
-    check_bond_dims, length, eachindex
+@forward PeriodicTensorTrain.tensors Base.getindex, Base.iterate, Base.firstindex, 
+    Base.lastindex, Base.setindex!, Base.length, Base.eachindex, 
+    check_bond_dims
 
 """
     flat_periodic_tt(bondsizes::AbstractVector{<:Integer}, q...)

diff --git a/src/tensor_train.jl b/src/tensor_train.jl
@@ -22,8 +22,9 @@ function TensorTrain(tensors::Vector{Array{F,N}}) where {F<:Number, N}
 end
 
 
-@forward TensorTrain.tensors getindex, iterate, firstindex, lastindex, setindex!, 
-    check_bond_dims, length, eachindex
+@forward TensorTrain.tensors Base.getindex, Base.iterate, Base.firstindex, Base.lastindex,
+    Base.setindex!, Base.length, Base.eachindex,  
+    check_bond_dims