[ITensors] Observers.jl package extension

.github/workflows/comment_trigger_test_itensortdvp.yml

@@ -51,6 +51,8 @@ jobs:
         run: |
           using Pkg;
           Pkg.add("ITensorTDVP");
+          Pkg.activate(temp=true);
+          Pkg.develop("ITensors");
           Pkg.test("ITensorTDVP");
       - name: Set latest commit status as ${{ job.status }}
         uses: myrotvorets/set-commit-status-action@master

.github/workflows/test_ndtensors.yml

@@ -38,3 +38,4 @@ jobs:
           Pkg.develop(path="./NDTensors");
           Pkg.develop(path=".");
           Pkg.test("NDTensors");
+

Project.toml

@@ -34,9 +34,11 @@ Zeros = "bd1ec220-6eb4-527a-9b49-e79c3db6233b"
 ZygoteRules = "700de1a5-db45-46bc-99cf-38207098b444"
 
 [weakdeps]
+Observers = "338f10d5-c7f1-4033-a7d1-f9dec39bcaa0"
 VectorInterface = "409d34a3-91d5-4945-b6ec-7529ddf182d8"
 
 [extensions]
+ITensorsObserversExt = "Observers"
 ITensorsVectorInterfaceExt = "VectorInterface"
 
 [compat]
@@ -72,4 +74,5 @@ ZygoteRules = "0.2.2"
 julia = "1.6"
 
 [extras]
+Observers = "338f10d5-c7f1-4033-a7d1-f9dec39bcaa0"
 VectorInterface = "409d34a3-91d5-4945-b6ec-7529ddf182d8"

ext/ITensorsObserversExt/ITensorsObserversExt.jl

@@ -0,0 +1,11 @@
+module ITensorsObserversExt
+
+using Observers: Observers
+using Observers.DataFrames: AbstractDataFrame
+using ITensors.ITensorMPS: ITensorMPS
+
+function ITensorMPS.update_observer!(observer::AbstractDataFrame; kwargs...)
+  return Observers.update!(observer; kwargs...)
+end
+
+end

src/ITensorMPS/alternating_update.jl

@@ -98,7 +98,7 @@ function alternating_update(
       )
     end
     current_time += time_step
-    update!(step_observer!; psi, sweep, outputlevel, current_time)
+    update_observer!(step_observer!; psi, sweep, outputlevel, current_time)
     if outputlevel >= 1
       print("After sweep ", sweep, ":")
       print(" maxlinkdim=", maxlinkdim(psi))