From aae6452825902f19aab4b0386343970a9a45b2e6 Mon Sep 17 00:00:00 2001
From: Johanna <johanna@zijderveld.de>
Date: Fri, 12 Apr 2024 12:30:29 +0200
Subject: [PATCH] Switch to using vectorized tb2kham version
---
codes/model.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/codes/model.py b/codes/model.py
index 69e1a4f..60c7e69 100644
--- a/codes/model.py
+++ b/codes/model.py
@@ -1,6 +1,6 @@
# %%
from codes.tb.tb import addTb
-from codes.tb.transforms import tb2kham, ifftn2tb
+from codes.tb.transforms import tb2khamvector, ifftn2tb
from codes.mf import (
densityMatrix,
fermiOnGrid,
@@ -36,7 +36,7 @@ class Model:
self.EF = fermiOnGrid(self.kham, self.filling)
def makeDensityMatrixTb(self, mf_model, nK=200):
- self.kham = tb2kham(addTb(self.h_0, mf_model), nK=nK, ndim=self._ndim)
+ self.kham = tb2khamvector(addTb(self.h_0, mf_model), nK=nK, ndim=self._ndim)
self.calculateEF()
return ifftn2tb(
ifftn(densityMatrix(self.kham, self.EF), axes=np.arange(self._ndim))
--
GitLab