Added truncate_kwargs parameter

corbett5 · corbett5 · commit 704046282112 · 2024-10-09T11:09:02.000-06:00
diff --git a/src/lib/ITensorMPS/src/mpo.jl b/src/lib/ITensorMPS/src/mpo.jl
@@ -826,6 +826,7 @@ function ITensors.contract(
   cutoff=1e-14,
   maxdim=maxlinkdim(A) * maxlinkdim(B),
   mindim=1,
+  truncate_kwargs=(;),
   kwargs...,
 )
   if hassameinds(siteinds, A, B)
@@ -874,6 +875,7 @@ function ITensors.contract(
     mindim,
     kwargs...,
   )
+  truncate!(C, truncate_kwargs...)
   return C
 end
 

Original file line number	Diff line number	Diff line change
`@@ -826,6 +826,7 @@ function ITensors.contract(`
`826`	`826`	`cutoff=1e-14,`
`827`	`827`	`maxdim=maxlinkdim(A) * maxlinkdim(B),`
`828`	`828`	`mindim=1,`
	`829`	`+ truncate_kwargs=(;),`
`829`	`830`	`kwargs...,`
`830`	`831`	`)`
`831`	`832`	`if hassameinds(siteinds, A, B)`
`@@ -874,6 +875,7 @@ function ITensors.contract(`
`874`	`875`	`mindim,`
`875`	`876`	`kwargs...,`
`876`	`877`	`)`
	`878`	`+ truncate!(C, truncate_kwargs...)`
`877`	`879`	`return C`
`878`	`880`	`end`
`879`	`881`