relax deduce_pspaces

Author: VictorVanthilt

src/operators/mpoham.jl

@@ -83,9 +83,18 @@ function deduce_pspaces(opps::SumOfLocalOperators)
     end
 
     non_deduced = map(ismissing, pspaces)
-    any(non_deduced) &&
-        error("cannot automatically deduce physical spaces at $(findall(non_deduced))")
 
+    if any(non_deduced) # Some spaces were not defined / not able to be deduced
+        if allequal(filter(!ismissing, pspaces)) # all non-missing spaces are equal
+            # fill in the missing spaces with the unique non-missing space
+            uniquespace = first(filter(!ismissing, pspaces))
+            for i in eachindex(pspaces)
+                pspaces[i] = uniquespace
+            end
+        else # Not all non-missing spaces are equal
+            error("cannot automatically deduce physical spaces at $(findall(non_deduced))")
+        end
+    end
     return collect(S, pspaces)
 end
 

test/mpoham.jl

@@ -1,4 +1,4 @@
-using MPSKitModels
+using MPSKitModels, MPSKit, TensorKit
 
 lattice = InfiniteChain(1)
 H1 = @mpoham begin
@@ -20,3 +20,13 @@ end
     H = @mpoham sum(ZZ{i,j} for (i, j) in nearest_neighbours(lattice))
     @test length(H) == length(lattice)
 end
+
+@testset "deduce_pspaces" begin
+    # Not fully defining the pspaces should still work
+    lattice = FiniteChain(5)
+    H = @mpoham S_zz(){lattice[2],lattice[3]}
+
+    @test unique(MPSKit.physicalspace(H))[1] == ComplexSpace(2)
+
+    @test_throws Exception @mpoham σˣ(){lattice[1]} + σˣ(; spin=3 // 2){lattice[2]}
+end