Improving infinite mpo matrix

src/ITensorInfiniteMPS.jl

@@ -34,9 +34,9 @@ include("itensormap.jl")
 include("celledvectors.jl")
 include("abstractinfinitemps.jl")
 include("infinitemps.jl")
-include("infinitempo.jl")
 include("infinitecanonicalmps.jl")
-include("infinitempomatrix.jl")
+include("infiniteblockmpo.jl")
+include("infinitempo.jl")
 include("transfermatrix.jl")
 include("models/models.jl")
 include("models/fqhe13.jl")
@@ -60,7 +60,7 @@ export Cell,
   InfMPS,
   InfiniteSum,
   InfiniteMPO,
-  InfiniteMPOMatrix,
+  InfiniteBlockMPO,
   InfiniteSumLocalOps,
   ITensorMap,
   ITensorNetwork,

src/ITensors.jl

@@ -182,3 +182,13 @@ end
 # TODO: make this definition AbstractMPS
 # Handle orthogonality center correctly
 Base.getindex(ψ::MPS, r::UnitRange{Int}) = MPS([ψ[n] for n in r])
+
+#TODO Remove if everything is working nicely
+#Was still crashing on my laptop after updating ITensors
+Base.fill!(::NDTensors.NoData, ::Any) = NDTensors.NoData()
+
+function ITensors.NDTensors.contraction_output(
+  A::NDTensors.EmptyTensor, B::NDTensors.DiagBlockSparseTensor, label
+)
+  return NDTensors.EmptyTensor(promote_type(eltype(A), eltype(B)), label)
+end

src/celledvectors.jl

@@ -142,6 +142,7 @@ _setindex_cell1!(cv::CelledVector, val, n::Int) = (ITensors.data(cv)[n] = val)
 function getindex(cv::CelledVector, n::Int)
   cellₙ = cell(cv, n)
   siteₙ = cellindex(cv, n)
+  cellₙ == 1 && return _getindex_cell1(cv, siteₙ) #Avoid unnecessary calls
   return translatecell(cv.translator, _getindex_cell1(cv, siteₙ), cellₙ - 1)
 end
 
@@ -169,7 +170,11 @@ getindex(cv::CelledVector, c::Cell) = cv[eachindex(cv, c)]
 function setindex!(cv::CelledVector, T, n::Int)
   cellₙ = cell(cv, n)
   siteₙ = cellindex(cv, n)
-  _setindex_cell1!(cv, translatecell(cv.translator, T, -(cellₙ - 1)), siteₙ)
+  if cellₙ == 1
+    _setindex_cell1!(cv, T, siteₙ)
+  else
+    _setindex_cell1!(cv, translatecell(cv.translator, T, -(cellₙ - 1)), siteₙ)
+  end
   return cv
 end