Renyi2 as optional

netket · Oct 31, 2023 · 8dea0f4 · 8dea0f4
1 parent c6356c6
commit 8dea0f4
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 4 deletions.
diff --git a/examples/dynamics_with_measurements/S2_dynamics_with_measurements.py b/examples/dynamics_with_measurements/S2_dynamics_with_measurements.py
@@ -7,6 +7,7 @@
 import jax
 
 from RBM_Jastrow_measurement import RBMJasMeas
+from netket_fidelity.renyi2 import Renyi2EntanglementEntropy
 
 # Set the parameters
 L = 2
@@ -64,7 +65,7 @@
 
 # Instantiate the Renyi2 entropy to monitor
 subsys = [x for x in range(N // 2)]
-S2op = nkf.Renyi2EntanglementEntropy(hi, subsys)
+S2op = Renyi2EntanglementEntropy(hi, subsys)
 
 
 # Compute the probabilities for the measurement outcomes of a spin

diff --git a/netket_fidelity/__init__.py b/netket_fidelity/__init__.py
@@ -5,8 +5,7 @@
 from . import driver
 
 from .infidelity import InfidelityOperator
-from .renyi2 import Renyi2EntanglementEntropy
 
 from netket.utils import _hide_submodules
 
-_hide_submodules(__name__, hide_folder=["renyi2", "infidelity"])
+_hide_submodules(__name__, hide_folder=["infidelity"])
diff --git a/test/test_Renyi2.py b/test/test_Renyi2.py
@@ -4,6 +4,7 @@
 import netket_fidelity as nkf
 
 from ._Renyi2_exact import _Renyi2_exact
+from netket_fidelity.renyi2 import Renyi2EntanglementEntropy
 
 N = 3
 hi = nk.hilbert.Spin(0.5, N)
@@ -29,7 +30,7 @@ def _setup():
     )
 
     subsys = [0, 1]
-    S2 = nkf.Renyi2EntanglementEntropy(hi, subsys)
+    S2 = Renyi2EntanglementEntropy(hi, subsys)
 
     return vs, vs_exact, S2, subsys