swap to abstractobservables

netket · Mar 3, 2024 · 0832c3a · 0832c3a
1 parent e5319d8
commit 0832c3a
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/netket_fidelity/infidelity/overlap/operator.py b/netket_fidelity/infidelity/overlap/operator.py
@@ -2,6 +2,7 @@
 
 import jax.numpy as jnp
 
+from netket.experimental.observable import AbstractObservable
 
 from netket.operator import AbstractOperator, DiscreteJaxOperator
 from netket.utils.types import DType
@@ -11,7 +12,7 @@
 from netket_fidelity.utils.sampling_Ustate import make_logpsi_U_afun
 
 
-class InfidelityOperatorStandard(AbstractOperator):
+class InfidelityOperatorStandard(AbstractObservable):
     def __init__(
         self,
         target: VariationalState,

diff --git a/netket_fidelity/infidelity/overlap_U/operator.py b/netket_fidelity/infidelity/overlap_U/operator.py
@@ -1,13 +1,15 @@
 from typing import Optional
 import jax.numpy as jnp
 
+from netket.experimental.observable import AbstractObservable
+
 from netket.operator import AbstractOperator
 from netket.utils.types import DType
 from netket.utils.numbers import is_scalar
 from netket.vqs import VariationalState, FullSumState
 
 
-class InfidelityOperatorUPsi(AbstractOperator):
+class InfidelityOperatorUPsi(AbstractObservable):
     def __init__(
         self,
         U: AbstractOperator,