Fix Base.similar for physical and contractible network types

src/networks/infinitesquarenetwork.jl

@@ -30,8 +30,8 @@
 Base.eltype(::Type{InfiniteSquareNetwork{O}}) where {O} = O
 
 Base.copy(n::InfiniteSquareNetwork) = InfiniteSquareNetwork(copy(unitcell(n)))
-function Base.similar(n::InfiniteSquareNetwork, args...)
-    return InfiniteSquareNetwork(similar(unitcell(n), args...))
+function Base.similar(n::InfiniteSquareNetwork, T::Type{TorA}=scalartype(n)) where {TorA}
+    return InfiniteSquareNetwork(map(t -> similar(t, T), unitcell(n)))
 end
 function Base.repeat(n::InfiniteSquareNetwork, counts...)
     return InfiniteSquareNetwork(repeat(unitcell(n), counts...))
@@ -53,6 +53,7 @@
 virtualspace(n::InfiniteSquareNetwork, r::Int, c::Int, dir) = virtualspace(n[r, c], dir)
 
 ## Vector interface
+
 function VectorInterface.scalartype(::Type{T}) where {T<:InfiniteSquareNetwork}
     return scalartype(eltype(T))
 end

src/operators/infinitepepo.jl

@@ -131,7 +131,9 @@
 Base.eltype(A::InfinitePEPO) = eltype(typeof(A))
 
 Base.copy(A::InfinitePEPO) = InfinitePEPO(copy(unitcell(A)))
-Base.similar(A::InfinitePEPO, args...) = InfinitePEPO(similar(unitcell(A), args...))
+function Base.similar(A::InfinitePEPO, T::Type{TorA}=scalartype(A)) where {TorA}
+    return InfinitePEPO(map(t -> similar(t, T), unitcell(A)))
+end
 Base.repeat(A::InfinitePEPO, counts...) = InfinitePEPO(repeat(unitcell(A), counts...))
 
 Base.getindex(A::InfinitePEPO, args...) = Base.getindex(unitcell(A), args...)
@@ -169,6 +171,15 @@
     )
 end
 
+## Vector interface
+
+function VectorInterface.scalartype(::Type{NT}) where {NT<:InfinitePEPO}
+    return scalartype(eltype(NT))
+end
+function VectorInterface.zerovector(A::InfinitePEPO)
+    return InfinitePEPO(zerovector(unitcell(A)))
+end
+
 ## (Approximate) equality
 function Base.:(==)(A₁::InfinitePEPO, A₂::InfinitePEPO)
     return all(zip(unitcell(A₁), unitcell(A₂))) do (p₁, p₂)

src/states/infinitepartitionfunction.jl

@@ -120,8 +120,10 @@
 Base.eltype(A::InfinitePartitionFunction) = eltype(typeof(A))
 
 Base.copy(A::InfinitePartitionFunction) = InfinitePartitionFunction(copy(unitcell(A)))
-function Base.similar(A::InfinitePartitionFunction, args...)
-    return InfinitePartitionFunction(similar(unitcell(A), args...))
+function Base.similar(
+    A::InfinitePartitionFunction, T::Type{TorA}=scalartype(A)
+) where {TorA}
+    return InfinitePartitionFunction(map(t -> similar(t, T), unitcell(A)))
 end
 function Base.repeat(A::InfinitePartitionFunction, counts...)
     return InfinitePartitionFunction(repeat(unitcell(A), counts...))
@@ -147,6 +149,15 @@
     return InfiniteSquareNetwork(unitcell(state))
 end
 
+## Vector interface
+
+function VectorInterface.scalartype(::Type{NT}) where {NT<:InfinitePartitionFunction}
+    return scalartype(eltype(NT))
+end
+function VectorInterface.zerovector(A::InfinitePartitionFunction)
+    return InfinitePartitionFunction(zerovector(unitcell(A)))
+end
+
 ## (Approximate) equality
 function Base.:(==)(A₁::InfinitePartitionFunction, A₂::InfinitePartitionFunction)
     return all(zip(unitcell(A₁), unitcell(A₂))) do (p₁, p₂)

src/states/infinitepeps.jl

@@ -122,7 +122,9 @@
 Base.eltype(A::InfinitePEPS) = eltype(typeof(A))
 
 Base.copy(A::InfinitePEPS) = InfinitePEPS(copy(unitcell(A)))
-Base.similar(A::InfinitePEPS, args...) = InfinitePEPS(similar(unitcell(A), args...))
+function Base.similar(A::InfinitePEPS, T::Type{TorA}=scalartype(A)) where {TorA}
+    return InfinitePEPS(map(t -> similar(t, T), unitcell(A)))
+end
 Base.repeat(A::InfinitePEPS, counts...) = InfinitePEPS(repeat(unitcell(A), counts...))
 
 Base.getindex(A::InfinitePEPS, args...) = Base.getindex(unitcell(A), args...)