From e311d44daa34b6f7cf34c673c5af7ee91eb293de Mon Sep 17 00:00:00 2001
From: Hao Zhang <hzhangxyz@outlook.com>
Date: Fri, 17 Nov 2023 18:57:52 +0800
Subject: [PATCH] Replace all "import TAT" to "from tat import compat as TAT".

---
 tetragono/tetragono/common_tensor/Fermi.py                      | 2 +-
 tetragono/tetragono/common_tensor/FermiFermi_Hubbard.py         | 2 +-
 tetragono/tetragono/common_tensor/FermiU1_Hubbard.py            | 2 +-
 tetragono/tetragono/common_tensor/FermiU1_tJ.py                 | 2 +-
 tetragono/tetragono/common_tensor/No.py                         | 2 +-
 tetragono/tetragono/common_tensor/Parity.py                     | 2 +-
 tetragono/tetragono/sampling_lattice/gradient.py                | 2 +-
 tetragono/tetragono/sampling_lattice/sampling.py                | 2 +-
 tetragono/tetragono/shell.py                                    | 2 +-
 tetragono/tetragono/utility.py                                  | 2 +-
 tetraku/tetraku/ansatzes/lx_cnn/__init__.py                     | 2 +-
 tetraku/tetraku/ansatzes/lx_cnn/pbc/__init__.py                 | 2 +-
 tetraku/tetraku/ansatzes/lx_cnn/pbc/deep.py                     | 2 +-
 tetraku/tetraku/models/J1J2/__init__.py                         | 2 +-
 tetraku/tetraku/models/J1J2/pbc/__init__.py                     | 2 +-
 tetraku/tetraku/models/bcs/square/d_wave.py                     | 2 +-
 tetraku/tetraku/models/bcs/square/s_wave.py                     | 2 +-
 tetraku/tetraku/models/boson_metal/__init__.py                  | 2 +-
 tetraku/tetraku/models/boson_metal/hopping_hamiltonian.py       | 2 +-
 tetraku/tetraku/models/boson_metal/initial_state.py             | 2 +-
 tetraku/tetraku/models/cluster_heisenberg/__init__.py           | 2 +-
 tetraku/tetraku/models/cluster_heisenberg/pbc/__init__.py       | 2 +-
 tetraku/tetraku/models/free_fermion/__init__.py                 | 2 +-
 tetraku/tetraku/models/free_fermion/complex/__init__.py         | 2 +-
 tetraku/tetraku/models/gibbs_free_fermion/__init__.py           | 2 +-
 .../tetraku/models/gibbs_free_fermion/hopping_hamiltonian.py    | 2 +-
 tetraku/tetraku/models/gibbs_heisenberg/__init__.py             | 2 +-
 tetraku/tetraku/models/gibbs_heisenberg/hopping_hamiltonian.py  | 2 +-
 .../gibbs_heisenberg/random_configuration_initialization.py     | 2 +-
 .../gibbs_heisenberg/trivial_configuration_initialization.py    | 2 +-
 tetraku/tetraku/models/gibbs_hubbard/__init__.py                | 2 +-
 .../half_filling_random_configuration_initialization.py         | 2 +-
 .../models/gibbs_hubbard/hopping_hamiltonian/__init__.py        | 2 +-
 .../gibbs_hubbard/hopping_hamiltonian/fix_particle_number.py    | 2 +-
 .../gibbs_hubbard/trivial_configuration_initialization.py       | 2 +-
 tetraku/tetraku/models/heisenberg/__init__.py                   | 2 +-
 tetraku/tetraku/models/heisenberg/complex/__init__.py           | 2 +-
 tetraku/tetraku/models/heisenberg/neel.py                       | 2 +-
 tetraku/tetraku/models/heisenberg/pbc/__init__.py               | 2 +-
 tetraku/tetraku/models/honeycomb_hubbard/__init__.py            | 2 +-
 tetraku/tetraku/models/honeycomb_hubbard/complex/__init__.py    | 2 +-
 tetraku/tetraku/models/honeycomb_hubbard/parity/__init__.py     | 2 +-
 .../models/honeycomb_hubbard/parity/initial_configuration.py    | 2 +-
 tetraku/tetraku/models/hubbard/__init__.py                      | 2 +-
 tetraku/tetraku/models/hubbard/complex/__init__.py              | 2 +-
 tetraku/tetraku/models/hubbard/fermi_fermi.py                   | 2 +-
 tetraku/tetraku/models/hubbard/fermi_u1.py                      | 2 +-
 tetraku/tetraku/models/hubbard/parity.py                        | 2 +-
 tetraku/tetraku/models/ising/__init__.py                        | 2 +-
 tetraku/tetraku/models/kagome_heisenberg/__init__.py            | 2 +-
 tetraku/tetraku/models/kagome_rydberg/__init__.py               | 2 +-
 tetraku/tetraku/models/kitaev/__init__.py                       | 2 +-
 tetraku/tetraku/models/kitaev/complex/__init__.py               | 2 +-
 tetraku/tetraku/models/rydberg/__init__.py                      | 2 +-
 tetraku/tetraku/models/rydberg/pbc/__init__.py                  | 2 +-
 tetraku/tetraku/models/rydberg/star/__init__.py                 | 2 +-
 tetraku/tetraku/models/tJ/__init__.py                           | 2 +-
 tetraku/tetraku/models/tV/__init__.py                           | 2 +-
 tetraku/tetraku/models/tV/parity/__init__.py                    | 2 +-
 .../tetraku/models/tV/parity/initial_half_particle_number.py    | 2 +-
 tetraku/tetraku/models/toric_code/__init__.py                   | 2 +-
 61 files changed, 61 insertions(+), 61 deletions(-)

diff --git a/tetragono/tetragono/common_tensor/Fermi.py b/tetragono/tetragono/common_tensor/Fermi.py
index c785cbb26..8ba4be15e 100644
--- a/tetragono/tetragono/common_tensor/Fermi.py
+++ b/tetragono/tetragono/common_tensor/Fermi.py
@@ -16,7 +16,7 @@
 # along with this program.  If not, see <https://www.gnu.org/licenses/>.
 #
 
-import TAT
+from tat import compat as TAT
 from .tensor_toolkit import Fedge, Tedge, rename_io
 
 Tensor = TAT.Fermi.Z.Tensor
diff --git a/tetragono/tetragono/common_tensor/FermiFermi_Hubbard.py b/tetragono/tetragono/common_tensor/FermiFermi_Hubbard.py
index 42f83dfdd..273fd2224 100644
--- a/tetragono/tetragono/common_tensor/FermiFermi_Hubbard.py
+++ b/tetragono/tetragono/common_tensor/FermiFermi_Hubbard.py
@@ -16,7 +16,7 @@
 # along with this program.  If not, see <https://www.gnu.org/licenses/>.
 #
 
-import TAT
+from tat import compat as TAT
 from .tensor_toolkit import Fedge, Tedge, rename_io
 
 Tensor = TAT.FermiFermi.Z.Tensor
diff --git a/tetragono/tetragono/common_tensor/FermiU1_Hubbard.py b/tetragono/tetragono/common_tensor/FermiU1_Hubbard.py
index f32b070cb..0ffca7794 100644
--- a/tetragono/tetragono/common_tensor/FermiU1_Hubbard.py
+++ b/tetragono/tetragono/common_tensor/FermiU1_Hubbard.py
@@ -16,7 +16,7 @@
 # along with this program.  If not, see <https://www.gnu.org/licenses/>.
 #
 
-import TAT
+from tat import compat as TAT
 from .tensor_toolkit import Fedge, Tedge, rename_io
 
 Tensor = TAT.FermiU1.Z.Tensor
diff --git a/tetragono/tetragono/common_tensor/FermiU1_tJ.py b/tetragono/tetragono/common_tensor/FermiU1_tJ.py
index 96284210c..7a02bd81a 100644
--- a/tetragono/tetragono/common_tensor/FermiU1_tJ.py
+++ b/tetragono/tetragono/common_tensor/FermiU1_tJ.py
@@ -16,7 +16,7 @@
 # along with this program.  If not, see <https://www.gnu.org/licenses/>.
 #
 
-import TAT
+from tat import compat as TAT
 from .tensor_toolkit import Fedge, Tedge, rename_io
 
 Tensor = TAT.FermiU1.Z.Tensor
diff --git a/tetragono/tetragono/common_tensor/No.py b/tetragono/tetragono/common_tensor/No.py
index 1d4c5913a..b5927a69a 100644
--- a/tetragono/tetragono/common_tensor/No.py
+++ b/tetragono/tetragono/common_tensor/No.py
@@ -16,7 +16,7 @@
 # along with this program.  If not, see <https://www.gnu.org/licenses/>.
 #
 
-import TAT
+from tat import compat as TAT
 from .tensor_toolkit import rename_io, kronecker_product
 
 Tensor = TAT.No.Z.Tensor
diff --git a/tetragono/tetragono/common_tensor/Parity.py b/tetragono/tetragono/common_tensor/Parity.py
index a30ee0d04..61c793066 100644
--- a/tetragono/tetragono/common_tensor/Parity.py
+++ b/tetragono/tetragono/common_tensor/Parity.py
@@ -16,7 +16,7 @@
 # along with this program.  If not, see <https://www.gnu.org/licenses/>.
 #
 
-import TAT
+from tat import compat as TAT
 from .tensor_toolkit import Fedge, Tedge, rename_io
 
 Tensor = TAT.Parity.Z.Tensor
diff --git a/tetragono/tetragono/sampling_lattice/gradient.py b/tetragono/tetragono/sampling_lattice/gradient.py
index f48fc59ad..d7ab6ae07 100644
--- a/tetragono/tetragono/sampling_lattice/gradient.py
+++ b/tetragono/tetragono/sampling_lattice/gradient.py
@@ -19,7 +19,7 @@
 import signal
 from datetime import datetime
 import numpy as np
-import TAT
+from tat import compat as TAT
 from ..sampling_lattice import SamplingLattice, Observer, SweepSampling, ErgodicSampling, DirectSampling
 from ..utility import (show, showln, mpi_rank, mpi_size, SignalHandler, seed_differ, lattice_randomize, write_to_file,
                        get_imported_function, restrict_wrapper, bcast_number, write_configurations)
diff --git a/tetragono/tetragono/sampling_lattice/sampling.py b/tetragono/tetragono/sampling_lattice/sampling.py
index 7ed769ad8..593261d8d 100644
--- a/tetragono/tetragono/sampling_lattice/sampling.py
+++ b/tetragono/tetragono/sampling_lattice/sampling.py
@@ -17,7 +17,7 @@
 #
 
 import numpy as np
-import TAT
+from tat import compat as TAT
 from ..auxiliaries import DoubleLayerAuxiliaries, ThreeLineAuxiliaries
 from ..utility import mpi_rank, mpi_size
 from ..tensor_element import tensor_element
diff --git a/tetragono/tetragono/shell.py b/tetragono/tetragono/shell.py
index 605605e34..6cc38fc89 100644
--- a/tetragono/tetragono/shell.py
+++ b/tetragono/tetragono/shell.py
@@ -21,7 +21,7 @@
 import cmd
 from io import StringIO
 import numpy as np
-import TAT
+from tat import compat as TAT
 from .utility import (mpi_rank, mpi_size, mpi_comm, write_to_file, read_from_file, show, showln, seed_differ,
                       get_imported_function, allgather_array, restrict_wrapper, write_configurations,
                       read_configurations)
diff --git a/tetragono/tetragono/utility.py b/tetragono/tetragono/utility.py
index ddd734688..2a0ced1dc 100644
--- a/tetragono/tetragono/utility.py
+++ b/tetragono/tetragono/utility.py
@@ -31,7 +31,7 @@
 import importlib.machinery
 from mpi4py import MPI
 import numpy as np
-import TAT
+from tat import compat as TAT
 
 clear_line = "\u001b[2K"
 
diff --git a/tetraku/tetraku/ansatzes/lx_cnn/__init__.py b/tetraku/tetraku/ansatzes/lx_cnn/__init__.py
index 4a24ee136..56ae841c5 100644
--- a/tetraku/tetraku/ansatzes/lx_cnn/__init__.py
+++ b/tetraku/tetraku/ansatzes/lx_cnn/__init__.py
@@ -17,7 +17,7 @@
 #
 
 import torch
-import TAT
+from tat import compat as TAT
 import tetragono as tet
 
 
diff --git a/tetraku/tetraku/ansatzes/lx_cnn/pbc/__init__.py b/tetraku/tetraku/ansatzes/lx_cnn/pbc/__init__.py
index b708b6bb3..b7247e12b 100644
--- a/tetraku/tetraku/ansatzes/lx_cnn/pbc/__init__.py
+++ b/tetraku/tetraku/ansatzes/lx_cnn/pbc/__init__.py
@@ -17,7 +17,7 @@
 #
 
 import torch
-import TAT
+from tat import compat as TAT
 import tetragono as tet
 
 
diff --git a/tetraku/tetraku/ansatzes/lx_cnn/pbc/deep.py b/tetraku/tetraku/ansatzes/lx_cnn/pbc/deep.py
index 533cb0040..671c074a8 100644
--- a/tetraku/tetraku/ansatzes/lx_cnn/pbc/deep.py
+++ b/tetraku/tetraku/ansatzes/lx_cnn/pbc/deep.py
@@ -17,7 +17,7 @@
 #
 
 import torch
-import TAT
+from tat import compat as TAT
 import tetragono as tet
 
 
diff --git a/tetraku/tetraku/models/J1J2/__init__.py b/tetraku/tetraku/models/J1J2/__init__.py
index e7489f659..2eb0ed7a8 100644
--- a/tetraku/tetraku/models/J1J2/__init__.py
+++ b/tetraku/tetraku/models/J1J2/__init__.py
@@ -16,7 +16,7 @@
 # along with this program.  If not, see <https://www.gnu.org/licenses/>.
 #
 
-import TAT
+from tat import compat as TAT
 import tetragono as tet
 
 
diff --git a/tetraku/tetraku/models/J1J2/pbc/__init__.py b/tetraku/tetraku/models/J1J2/pbc/__init__.py
index 5ba8f6413..f3ad0713b 100644
--- a/tetraku/tetraku/models/J1J2/pbc/__init__.py
+++ b/tetraku/tetraku/models/J1J2/pbc/__init__.py
@@ -16,7 +16,7 @@
 # along with this program.  If not, see <https://www.gnu.org/licenses/>.
 #
 
-import TAT
+from tat import compat as TAT
 import tetragono as tet
 
 
diff --git a/tetraku/tetraku/models/bcs/square/d_wave.py b/tetraku/tetraku/models/bcs/square/d_wave.py
index 53041e8ee..ef2efa8eb 100644
--- a/tetraku/tetraku/models/bcs/square/d_wave.py
+++ b/tetraku/tetraku/models/bcs/square/d_wave.py
@@ -16,7 +16,7 @@
 # along with this program.  If not, see <https://www.gnu.org/licenses/>.
 #
 
-import TAT
+from tat import compat as TAT
 import tetragono as tet
 
 
diff --git a/tetraku/tetraku/models/bcs/square/s_wave.py b/tetraku/tetraku/models/bcs/square/s_wave.py
index 915e57ee5..4050dc98c 100644
--- a/tetraku/tetraku/models/bcs/square/s_wave.py
+++ b/tetraku/tetraku/models/bcs/square/s_wave.py
@@ -16,7 +16,7 @@
 # along with this program.  If not, see <https://www.gnu.org/licenses/>.
 #
 
-import TAT
+from tat import compat as TAT
 import tetragono as tet
 
 
diff --git a/tetraku/tetraku/models/boson_metal/__init__.py b/tetraku/tetraku/models/boson_metal/__init__.py
index 79abd824b..18894e442 100644
--- a/tetraku/tetraku/models/boson_metal/__init__.py
+++ b/tetraku/tetraku/models/boson_metal/__init__.py
@@ -16,7 +16,7 @@
 # along with this program.  If not, see <https://www.gnu.org/licenses/>.
 #
 
-import TAT
+from tat import compat as TAT
 import tetragono as tet
 
 
diff --git a/tetraku/tetraku/models/boson_metal/hopping_hamiltonian.py b/tetraku/tetraku/models/boson_metal/hopping_hamiltonian.py
index 5a20b930b..55266c283 100644
--- a/tetraku/tetraku/models/boson_metal/hopping_hamiltonian.py
+++ b/tetraku/tetraku/models/boson_metal/hopping_hamiltonian.py
@@ -16,7 +16,7 @@
 # along with this program.  If not, see <https://www.gnu.org/licenses/>.
 #
 
-import TAT
+from tat import compat as TAT
 import tetragono as tet
 
 
diff --git a/tetraku/tetraku/models/boson_metal/initial_state.py b/tetraku/tetraku/models/boson_metal/initial_state.py
index f7673fc23..74182b3fb 100644
--- a/tetraku/tetraku/models/boson_metal/initial_state.py
+++ b/tetraku/tetraku/models/boson_metal/initial_state.py
@@ -16,7 +16,7 @@
 # along with this program.  If not, see <https://www.gnu.org/licenses/>.
 #
 
-import TAT
+from tat import compat as TAT
 import tetragono as tet
 
 
diff --git a/tetraku/tetraku/models/cluster_heisenberg/__init__.py b/tetraku/tetraku/models/cluster_heisenberg/__init__.py
index 3fdb733de..26d27566a 100644
--- a/tetraku/tetraku/models/cluster_heisenberg/__init__.py
+++ b/tetraku/tetraku/models/cluster_heisenberg/__init__.py
@@ -16,7 +16,7 @@
 # along with this program.  If not, see <https://www.gnu.org/licenses/>.
 #
 
-import TAT
+from tat import compat as TAT
 import tetragono as tet
 
 
diff --git a/tetraku/tetraku/models/cluster_heisenberg/pbc/__init__.py b/tetraku/tetraku/models/cluster_heisenberg/pbc/__init__.py
index e4cf13c49..3dfc14167 100644
--- a/tetraku/tetraku/models/cluster_heisenberg/pbc/__init__.py
+++ b/tetraku/tetraku/models/cluster_heisenberg/pbc/__init__.py
@@ -16,7 +16,7 @@
 # along with this program.  If not, see <https://www.gnu.org/licenses/>.
 #
 
-import TAT
+from tat import compat as TAT
 import tetragono as tet
 
 
diff --git a/tetraku/tetraku/models/free_fermion/__init__.py b/tetraku/tetraku/models/free_fermion/__init__.py
index 0faccbbbe..d0b39c08a 100644
--- a/tetraku/tetraku/models/free_fermion/__init__.py
+++ b/tetraku/tetraku/models/free_fermion/__init__.py
@@ -16,7 +16,7 @@
 # along with this program.  If not, see <https://www.gnu.org/licenses/>.
 #
 
-import TAT
+from tat import compat as TAT
 import tetragono as tet
 
 
diff --git a/tetraku/tetraku/models/free_fermion/complex/__init__.py b/tetraku/tetraku/models/free_fermion/complex/__init__.py
index 757f2b67b..9f0183542 100644
--- a/tetraku/tetraku/models/free_fermion/complex/__init__.py
+++ b/tetraku/tetraku/models/free_fermion/complex/__init__.py
@@ -16,7 +16,7 @@
 # along with this program.  If not, see <https://www.gnu.org/licenses/>.
 #
 
-import TAT
+from tat import compat as TAT
 import tetragono as tet
 
 
diff --git a/tetraku/tetraku/models/gibbs_free_fermion/__init__.py b/tetraku/tetraku/models/gibbs_free_fermion/__init__.py
index 22d18232a..502f5921d 100644
--- a/tetraku/tetraku/models/gibbs_free_fermion/__init__.py
+++ b/tetraku/tetraku/models/gibbs_free_fermion/__init__.py
@@ -16,7 +16,7 @@
 # along with this program.  If not, see <https://www.gnu.org/licenses/>.
 #
 
-import TAT
+from tat import compat as TAT
 import tetragono as tet
 from tetragono.common_tensor.tensor_toolkit import half_reverse
 
diff --git a/tetraku/tetraku/models/gibbs_free_fermion/hopping_hamiltonian.py b/tetraku/tetraku/models/gibbs_free_fermion/hopping_hamiltonian.py
index 98c0c78c0..9360386c9 100644
--- a/tetraku/tetraku/models/gibbs_free_fermion/hopping_hamiltonian.py
+++ b/tetraku/tetraku/models/gibbs_free_fermion/hopping_hamiltonian.py
@@ -17,7 +17,7 @@
 # along with this program.  If not, see <https://www.gnu.org/licenses/>.
 #
 
-import TAT
+from tat import compat as TAT
 import tetragono as tet
 from tetragono.common_tensor.Parity import *
 from tetragono.common_tensor.tensor_toolkit import rename_io, kronecker_product, half_reverse
diff --git a/tetraku/tetraku/models/gibbs_heisenberg/__init__.py b/tetraku/tetraku/models/gibbs_heisenberg/__init__.py
index 266a99ac9..e987d19c9 100644
--- a/tetraku/tetraku/models/gibbs_heisenberg/__init__.py
+++ b/tetraku/tetraku/models/gibbs_heisenberg/__init__.py
@@ -16,7 +16,7 @@
 # along with this program.  If not, see <https://www.gnu.org/licenses/>.
 #
 
-import TAT
+from tat import compat as TAT
 import tetragono as tet
 
 
diff --git a/tetraku/tetraku/models/gibbs_heisenberg/hopping_hamiltonian.py b/tetraku/tetraku/models/gibbs_heisenberg/hopping_hamiltonian.py
index 1effad39b..d66167f89 100644
--- a/tetraku/tetraku/models/gibbs_heisenberg/hopping_hamiltonian.py
+++ b/tetraku/tetraku/models/gibbs_heisenberg/hopping_hamiltonian.py
@@ -17,7 +17,7 @@
 # along with this program.  If not, see <https://www.gnu.org/licenses/>.
 #
 
-import TAT
+from tat import compat as TAT
 import tetragono as tet
 
 
diff --git a/tetraku/tetraku/models/gibbs_heisenberg/random_configuration_initialization.py b/tetraku/tetraku/models/gibbs_heisenberg/random_configuration_initialization.py
index 33d9f91e0..f74e21a8c 100644
--- a/tetraku/tetraku/models/gibbs_heisenberg/random_configuration_initialization.py
+++ b/tetraku/tetraku/models/gibbs_heisenberg/random_configuration_initialization.py
@@ -17,7 +17,7 @@
 # along with this program.  If not, see <https://www.gnu.org/licenses/>.
 #
 
-import TAT
+from tat import compat as TAT
 import tetragono as tet
 
 
diff --git a/tetraku/tetraku/models/gibbs_heisenberg/trivial_configuration_initialization.py b/tetraku/tetraku/models/gibbs_heisenberg/trivial_configuration_initialization.py
index 8ec91e386..938f6af28 100644
--- a/tetraku/tetraku/models/gibbs_heisenberg/trivial_configuration_initialization.py
+++ b/tetraku/tetraku/models/gibbs_heisenberg/trivial_configuration_initialization.py
@@ -17,7 +17,7 @@
 # along with this program.  If not, see <https://www.gnu.org/licenses/>.
 #
 
-import TAT
+from tat import compat as TAT
 import tetragono as tet
 
 
diff --git a/tetraku/tetraku/models/gibbs_hubbard/__init__.py b/tetraku/tetraku/models/gibbs_hubbard/__init__.py
index 5b81b847e..96be89cbd 100644
--- a/tetraku/tetraku/models/gibbs_hubbard/__init__.py
+++ b/tetraku/tetraku/models/gibbs_hubbard/__init__.py
@@ -16,7 +16,7 @@
 # along with this program.  If not, see <https://www.gnu.org/licenses/>.
 #
 
-import TAT
+from tat import compat as TAT
 import tetragono as tet
 from tetragono.common_tensor.tensor_toolkit import half_reverse
 
diff --git a/tetraku/tetraku/models/gibbs_hubbard/half_filling_random_configuration_initialization.py b/tetraku/tetraku/models/gibbs_hubbard/half_filling_random_configuration_initialization.py
index 8e47b86ce..109902193 100644
--- a/tetraku/tetraku/models/gibbs_hubbard/half_filling_random_configuration_initialization.py
+++ b/tetraku/tetraku/models/gibbs_hubbard/half_filling_random_configuration_initialization.py
@@ -17,7 +17,7 @@
 # along with this program.  If not, see <https://www.gnu.org/licenses/>.
 #
 
-import TAT
+from tat import compat as TAT
 import tetragono as tet
 
 
diff --git a/tetraku/tetraku/models/gibbs_hubbard/hopping_hamiltonian/__init__.py b/tetraku/tetraku/models/gibbs_hubbard/hopping_hamiltonian/__init__.py
index 36a5e3e84..aa8b92e80 100644
--- a/tetraku/tetraku/models/gibbs_hubbard/hopping_hamiltonian/__init__.py
+++ b/tetraku/tetraku/models/gibbs_hubbard/hopping_hamiltonian/__init__.py
@@ -17,7 +17,7 @@
 # along with this program.  If not, see <https://www.gnu.org/licenses/>.
 #
 
-import TAT
+from tat import compat as TAT
 import tetragono as tet
 from tetragono.common_tensor.Parity import *
 from tetragono.common_tensor.tensor_toolkit import rename_io, kronecker_product, half_reverse
diff --git a/tetraku/tetraku/models/gibbs_hubbard/hopping_hamiltonian/fix_particle_number.py b/tetraku/tetraku/models/gibbs_hubbard/hopping_hamiltonian/fix_particle_number.py
index 43429b6ee..a8d2147b0 100644
--- a/tetraku/tetraku/models/gibbs_hubbard/hopping_hamiltonian/fix_particle_number.py
+++ b/tetraku/tetraku/models/gibbs_hubbard/hopping_hamiltonian/fix_particle_number.py
@@ -17,7 +17,7 @@
 # along with this program.  If not, see <https://www.gnu.org/licenses/>.
 #
 
-import TAT
+from tat import compat as TAT
 import tetragono as tet
 from tetragono.common_tensor.Parity import *
 from tetragono.common_tensor.tensor_toolkit import rename_io, kronecker_product, half_reverse
diff --git a/tetraku/tetraku/models/gibbs_hubbard/trivial_configuration_initialization.py b/tetraku/tetraku/models/gibbs_hubbard/trivial_configuration_initialization.py
index 57083f572..26822dae5 100644
--- a/tetraku/tetraku/models/gibbs_hubbard/trivial_configuration_initialization.py
+++ b/tetraku/tetraku/models/gibbs_hubbard/trivial_configuration_initialization.py
@@ -17,7 +17,7 @@
 # along with this program.  If not, see <https://www.gnu.org/licenses/>.
 #
 
-import TAT
+from tat import compat as TAT
 import tetragono as tet
 
 
diff --git a/tetraku/tetraku/models/heisenberg/__init__.py b/tetraku/tetraku/models/heisenberg/__init__.py
index 548e855c9..422ef8405 100644
--- a/tetraku/tetraku/models/heisenberg/__init__.py
+++ b/tetraku/tetraku/models/heisenberg/__init__.py
@@ -16,7 +16,7 @@
 # along with this program.  If not, see <https://www.gnu.org/licenses/>.
 #
 
-import TAT
+from tat import compat as TAT
 import tetragono as tet
 
 
diff --git a/tetraku/tetraku/models/heisenberg/complex/__init__.py b/tetraku/tetraku/models/heisenberg/complex/__init__.py
index 929df1388..e5c02c086 100644
--- a/tetraku/tetraku/models/heisenberg/complex/__init__.py
+++ b/tetraku/tetraku/models/heisenberg/complex/__init__.py
@@ -16,7 +16,7 @@
 # along with this program.  If not, see <https://www.gnu.org/licenses/>.
 #
 
-import TAT
+from tat import compat as TAT
 import tetragono as tet
 
 
diff --git a/tetraku/tetraku/models/heisenberg/neel.py b/tetraku/tetraku/models/heisenberg/neel.py
index cc931c0e3..03a994092 100644
--- a/tetraku/tetraku/models/heisenberg/neel.py
+++ b/tetraku/tetraku/models/heisenberg/neel.py
@@ -16,7 +16,7 @@
 # along with this program.  If not, see <https://www.gnu.org/licenses/>.
 #
 
-import TAT
+from tat import compat as TAT
 
 
 def initial_configuration(conf, zero_first=True):
diff --git a/tetraku/tetraku/models/heisenberg/pbc/__init__.py b/tetraku/tetraku/models/heisenberg/pbc/__init__.py
index 5353b25ce..79f8cbd06 100644
--- a/tetraku/tetraku/models/heisenberg/pbc/__init__.py
+++ b/tetraku/tetraku/models/heisenberg/pbc/__init__.py
@@ -16,7 +16,7 @@
 # along with this program.  If not, see <https://www.gnu.org/licenses/>.
 #
 
-import TAT
+from tat import compat as TAT
 import tetragono as tet
 
 
diff --git a/tetraku/tetraku/models/honeycomb_hubbard/__init__.py b/tetraku/tetraku/models/honeycomb_hubbard/__init__.py
index dfe55f40f..ac2109e72 100644
--- a/tetraku/tetraku/models/honeycomb_hubbard/__init__.py
+++ b/tetraku/tetraku/models/honeycomb_hubbard/__init__.py
@@ -16,7 +16,7 @@
 # along with this program.  If not, see <https://www.gnu.org/licenses/>.
 #
 
-import TAT
+from tat import compat as TAT
 import tetragono as tet
 
 
diff --git a/tetraku/tetraku/models/honeycomb_hubbard/complex/__init__.py b/tetraku/tetraku/models/honeycomb_hubbard/complex/__init__.py
index 1af81fa67..e24e6c45c 100644
--- a/tetraku/tetraku/models/honeycomb_hubbard/complex/__init__.py
+++ b/tetraku/tetraku/models/honeycomb_hubbard/complex/__init__.py
@@ -16,7 +16,7 @@
 # along with this program.  If not, see <https://www.gnu.org/licenses/>.
 #
 
-import TAT
+from tat import compat as TAT
 import tetragono as tet
 
 
diff --git a/tetraku/tetraku/models/honeycomb_hubbard/parity/__init__.py b/tetraku/tetraku/models/honeycomb_hubbard/parity/__init__.py
index 03d09477a..eef44c019 100644
--- a/tetraku/tetraku/models/honeycomb_hubbard/parity/__init__.py
+++ b/tetraku/tetraku/models/honeycomb_hubbard/parity/__init__.py
@@ -16,7 +16,7 @@
 # along with this program.  If not, see <https://www.gnu.org/licenses/>.
 #
 
-import TAT
+from tat import compat as TAT
 import tetragono as tet
 
 
diff --git a/tetraku/tetraku/models/honeycomb_hubbard/parity/initial_configuration.py b/tetraku/tetraku/models/honeycomb_hubbard/parity/initial_configuration.py
index 35aca3933..db55f48f9 100644
--- a/tetraku/tetraku/models/honeycomb_hubbard/parity/initial_configuration.py
+++ b/tetraku/tetraku/models/honeycomb_hubbard/parity/initial_configuration.py
@@ -17,7 +17,7 @@
 # along with this program.  If not, see <https://www.gnu.org/licenses/>.
 #
 
-import TAT
+from tat import compat as TAT
 import tetragono as tet
 import os
 
diff --git a/tetraku/tetraku/models/hubbard/__init__.py b/tetraku/tetraku/models/hubbard/__init__.py
index fe9280104..060d7847e 100644
--- a/tetraku/tetraku/models/hubbard/__init__.py
+++ b/tetraku/tetraku/models/hubbard/__init__.py
@@ -16,7 +16,7 @@
 # along with this program.  If not, see <https://www.gnu.org/licenses/>.
 #
 
-import TAT
+from tat import compat as TAT
 import tetragono as tet
 
 
diff --git a/tetraku/tetraku/models/hubbard/complex/__init__.py b/tetraku/tetraku/models/hubbard/complex/__init__.py
index ae6726148..517c29c74 100644
--- a/tetraku/tetraku/models/hubbard/complex/__init__.py
+++ b/tetraku/tetraku/models/hubbard/complex/__init__.py
@@ -16,7 +16,7 @@
 # along with this program.  If not, see <https://www.gnu.org/licenses/>.
 #
 
-import TAT
+from tat import compat as TAT
 import tetragono as tet
 
 
diff --git a/tetraku/tetraku/models/hubbard/fermi_fermi.py b/tetraku/tetraku/models/hubbard/fermi_fermi.py
index 8ca5c89b3..c7863b422 100644
--- a/tetraku/tetraku/models/hubbard/fermi_fermi.py
+++ b/tetraku/tetraku/models/hubbard/fermi_fermi.py
@@ -16,7 +16,7 @@
 # along with this program.  If not, see <https://www.gnu.org/licenses/>.
 #
 
-import TAT
+from tat import compat as TAT
 import tetragono as tet
 
 
diff --git a/tetraku/tetraku/models/hubbard/fermi_u1.py b/tetraku/tetraku/models/hubbard/fermi_u1.py
index 77e376eed..e4ce7dcec 100644
--- a/tetraku/tetraku/models/hubbard/fermi_u1.py
+++ b/tetraku/tetraku/models/hubbard/fermi_u1.py
@@ -16,7 +16,7 @@
 # along with this program.  If not, see <https://www.gnu.org/licenses/>.
 #
 
-import TAT
+from tat import compat as TAT
 import tetragono as tet
 
 
diff --git a/tetraku/tetraku/models/hubbard/parity.py b/tetraku/tetraku/models/hubbard/parity.py
index e1a411a38..5068ccfef 100644
--- a/tetraku/tetraku/models/hubbard/parity.py
+++ b/tetraku/tetraku/models/hubbard/parity.py
@@ -16,7 +16,7 @@
 # along with this program.  If not, see <https://www.gnu.org/licenses/>.
 #
 
-import TAT
+from tat import compat as TAT
 import tetragono as tet
 
 
diff --git a/tetraku/tetraku/models/ising/__init__.py b/tetraku/tetraku/models/ising/__init__.py
index efb190431..0c858cbd8 100644
--- a/tetraku/tetraku/models/ising/__init__.py
+++ b/tetraku/tetraku/models/ising/__init__.py
@@ -16,7 +16,7 @@
 # along with this program.  If not, see <https://www.gnu.org/licenses/>.
 #
 
-import TAT
+from tat import compat as TAT
 import tetragono as tet
 
 
diff --git a/tetraku/tetraku/models/kagome_heisenberg/__init__.py b/tetraku/tetraku/models/kagome_heisenberg/__init__.py
index c84043715..6b131a13d 100644
--- a/tetraku/tetraku/models/kagome_heisenberg/__init__.py
+++ b/tetraku/tetraku/models/kagome_heisenberg/__init__.py
@@ -16,7 +16,7 @@
 # along with this program.  If not, see <https://www.gnu.org/licenses/>.
 #
 
-import TAT
+from tat import compat as TAT
 import tetragono as tet
 
 
diff --git a/tetraku/tetraku/models/kagome_rydberg/__init__.py b/tetraku/tetraku/models/kagome_rydberg/__init__.py
index 080486b03..1a55da193 100644
--- a/tetraku/tetraku/models/kagome_rydberg/__init__.py
+++ b/tetraku/tetraku/models/kagome_rydberg/__init__.py
@@ -16,7 +16,7 @@
 # along with this program.  If not, see <https://www.gnu.org/licenses/>.
 #
 
-import TAT
+from tat import compat as TAT
 import tetragono as tet
 from tetragono.common_tensor.tensor_toolkit import rename_io, kronecker_product
 
diff --git a/tetraku/tetraku/models/kitaev/__init__.py b/tetraku/tetraku/models/kitaev/__init__.py
index 0d925d08a..b5a696799 100644
--- a/tetraku/tetraku/models/kitaev/__init__.py
+++ b/tetraku/tetraku/models/kitaev/__init__.py
@@ -16,7 +16,7 @@
 # along with this program.  If not, see <https://www.gnu.org/licenses/>.
 #
 
-import TAT
+from tat import compat as TAT
 import tetragono as tet
 
 
diff --git a/tetraku/tetraku/models/kitaev/complex/__init__.py b/tetraku/tetraku/models/kitaev/complex/__init__.py
index 428195b68..50fe7892d 100644
--- a/tetraku/tetraku/models/kitaev/complex/__init__.py
+++ b/tetraku/tetraku/models/kitaev/complex/__init__.py
@@ -16,7 +16,7 @@
 # along with this program.  If not, see <https://www.gnu.org/licenses/>.
 #
 
-import TAT
+from tat import compat as TAT
 import tetragono as tet
 
 
diff --git a/tetraku/tetraku/models/rydberg/__init__.py b/tetraku/tetraku/models/rydberg/__init__.py
index bc8f16d3b..7af06c61e 100644
--- a/tetraku/tetraku/models/rydberg/__init__.py
+++ b/tetraku/tetraku/models/rydberg/__init__.py
@@ -16,7 +16,7 @@
 # along with this program.  If not, see <https://www.gnu.org/licenses/>.
 #
 
-import TAT
+from tat import compat as TAT
 import tetragono as tet
 from tetragono.common_tensor.tensor_toolkit import rename_io, kronecker_product
 
diff --git a/tetraku/tetraku/models/rydberg/pbc/__init__.py b/tetraku/tetraku/models/rydberg/pbc/__init__.py
index daa6428ac..d16b95493 100644
--- a/tetraku/tetraku/models/rydberg/pbc/__init__.py
+++ b/tetraku/tetraku/models/rydberg/pbc/__init__.py
@@ -16,7 +16,7 @@
 # along with this program.  If not, see <https://www.gnu.org/licenses/>.
 #
 
-import TAT
+from tat import compat as TAT
 import tetragono as tet
 from tetragono.common_tensor.tensor_toolkit import rename_io, kronecker_product
 
diff --git a/tetraku/tetraku/models/rydberg/star/__init__.py b/tetraku/tetraku/models/rydberg/star/__init__.py
index 5e8b1f159..bb1b26c35 100644
--- a/tetraku/tetraku/models/rydberg/star/__init__.py
+++ b/tetraku/tetraku/models/rydberg/star/__init__.py
@@ -16,7 +16,7 @@
 # along with this program.  If not, see <https://www.gnu.org/licenses/>.
 #
 
-import TAT
+from tat import compat as TAT
 import tetragono as tet
 from tetragono.common_tensor.tensor_toolkit import rename_io, kronecker_product
 
diff --git a/tetraku/tetraku/models/tJ/__init__.py b/tetraku/tetraku/models/tJ/__init__.py
index f63d66434..956c31768 100644
--- a/tetraku/tetraku/models/tJ/__init__.py
+++ b/tetraku/tetraku/models/tJ/__init__.py
@@ -16,7 +16,7 @@
 # along with this program.  If not, see <https://www.gnu.org/licenses/>.
 #
 
-import TAT
+from tat import compat as TAT
 import tetragono as tet
 
 
diff --git a/tetraku/tetraku/models/tV/__init__.py b/tetraku/tetraku/models/tV/__init__.py
index b1eb34ed4..785de283f 100644
--- a/tetraku/tetraku/models/tV/__init__.py
+++ b/tetraku/tetraku/models/tV/__init__.py
@@ -16,7 +16,7 @@
 # along with this program.  If not, see <https://www.gnu.org/licenses/>.
 #
 
-import TAT
+from tat import compat as TAT
 import tetragono as tet
 from tetragono.common_tensor.tensor_toolkit import rename_io, kronecker_product
 
diff --git a/tetraku/tetraku/models/tV/parity/__init__.py b/tetraku/tetraku/models/tV/parity/__init__.py
index 88af2db6a..57b6801f1 100644
--- a/tetraku/tetraku/models/tV/parity/__init__.py
+++ b/tetraku/tetraku/models/tV/parity/__init__.py
@@ -16,7 +16,7 @@
 # along with this program.  If not, see <https://www.gnu.org/licenses/>.
 #
 
-import TAT
+from tat import compat as TAT
 import tetragono as tet
 from tetragono.common_tensor.tensor_toolkit import rename_io, kronecker_product
 
diff --git a/tetraku/tetraku/models/tV/parity/initial_half_particle_number.py b/tetraku/tetraku/models/tV/parity/initial_half_particle_number.py
index 17b633b73..e44613c9d 100644
--- a/tetraku/tetraku/models/tV/parity/initial_half_particle_number.py
+++ b/tetraku/tetraku/models/tV/parity/initial_half_particle_number.py
@@ -17,7 +17,7 @@
 # along with this program.  If not, see <https://www.gnu.org/licenses/>.
 #
 
-import TAT
+from tat import compat as TAT
 import tetragono as tet
 
 
diff --git a/tetraku/tetraku/models/toric_code/__init__.py b/tetraku/tetraku/models/toric_code/__init__.py
index da76c9096..cc027404f 100644
--- a/tetraku/tetraku/models/toric_code/__init__.py
+++ b/tetraku/tetraku/models/toric_code/__init__.py
@@ -16,7 +16,7 @@
 # along with this program.  If not, see <https://www.gnu.org/licenses/>.
 #
 
-import TAT
+from tat import compat as TAT
 import tetragono as tet
 from tetragono.common_tensor.tensor_toolkit import rename_io, kronecker_product