From 233f7abfd81fb483710c64abbd4bc02248b71093 Mon Sep 17 00:00:00 2001
From: Antonio Manesco <am@antoniomanesco.org>
Date: Wed, 20 Dec 2023 09:44:46 +0100
Subject: [PATCH] fix k-space potential
---
codes/model.py | 9 +++++++--
1 file changed, 7 insertions(+), 2 deletions(-)
diff --git a/codes/model.py b/codes/model.py
index e566414..e48086a 100644
--- a/codes/model.py
+++ b/codes/model.py
@@ -6,7 +6,8 @@ class Model:
def __init__(self, tb_model, int_model=None, Vk=None, guess=None):
self.tb_model = tb_model
self.hk = utils.model2hk(tb_model=tb_model)
- if int_model is not None:
+ self.int_model = int_model
+ if self.int_model is not None:
self.int_model = int_model
self.Vk = utils.model2hk(tb_model=int_model)
else:
@@ -17,10 +18,14 @@ class Model:
def random_guess(self, vectors):
+ if self.int_model is None:
+ scale = 1
+ else:
+ scale = 1+np.max(np.abs([*self.int_model.values()]))
self.guess = utils.generate_guess(
vectors=vectors,
ndof=self.ndof,
- scale=1+np.max(np.abs([*self.int_model.values()]))
+ scale=scale
)
def kgrid_evaluation(self, nk):
--
GitLab