From 71a14e73db4373a4a81f447ee51e2502538a594f Mon Sep 17 00:00:00 2001
From: Antonio Manesco <am@antoniomanesco.org>
Date: Wed, 20 Dec 2023 09:45:05 +0100
Subject: [PATCH] adapt for k-space potential

---
 codes/hf.py | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/codes/hf.py b/codes/hf.py
index 4f76ea8..713345c 100644
--- a/codes/hf.py
+++ b/codes/hf.py
@@ -273,11 +273,13 @@ def find_groundstate_ham(
     """
     model.nk=nk
     model.filling=filling
-    vectors = utils.generate_vectors(cutoff_Vk, model.dim)
-    model.vectors=[*model.int_model.keys()]
+    if model.int_model is not None:
+        model.vectors=[*model.int_model.keys()]
+    else:
+        model.vectors = utils.generate_vectors(cutoff_Vk, model.dim)
     if model.guess is None:
         model.random_guess(model.vectors)
     solver(model, optimizer, optimizer_kwargs)
-    model.vectors=[*model.int_model, *model.tb_model.keys()]
+    model.vectors=[*model.vectors, *model.tb_model.keys()]
     assert np.allclose((model.mf_k - np.moveaxis(model.mf_k, -1, -2).conj())/2, 0, atol=1e-15)
     return utils.hk2tb_model(model.hamiltonians_0 + model.mf_k, model.vectors, model.ks)
-- 
GitLab