From a807dba921aba16f93b3a4eb768d9233c36da5c1 Mon Sep 17 00:00:00 2001
From: Kostas Vilkelis <kostasvilkelis@gmail.com>
Date: Tue, 7 May 2024 00:15:39 +0200
Subject: [PATCH] use ndof instead of size
---
pymf/model.py | 4 ++--
pymf/solvers.py | 6 +++---
2 files changed, 5 insertions(+), 5 deletions(-)
diff --git a/pymf/model.py b/pymf/model.py
index d5c169f..4ca80df 100644
--- a/pymf/model.py
+++ b/pymf/model.py
@@ -67,7 +67,7 @@ class Model:
_first_key = list(h_0)[0]
self._ndim = len(_first_key)
- self._size = h_0[_first_key].shape[0]
+ self._ndof = h_0[_first_key].shape[0]
self._local_key = tuple(np.zeros((self._ndim,), dtype=int))
_check_hermiticity(h_0)
@@ -94,5 +94,5 @@ class Model:
)
return add_tb(
meanfield(rho, self.h_int),
- {self._local_key: -fermi_energy * np.eye(self._size)},
+ {self._local_key: -fermi_energy * np.eye(self._ndof)},
)
diff --git a/pymf/solvers.py b/pymf/solvers.py
index 60d2db6..1796fcf 100644
--- a/pymf/solvers.py
+++ b/pymf/solvers.py
@@ -30,7 +30,7 @@ def cost(mf_param: np.ndarray, model: Model, nk: int = 100) -> np.ndarray:
1D real array that is the difference between the computed and inputted mean-field
parametrisations
"""
- shape = model._size
+ shape = model._ndof
mf = rparams_to_tb(mf_param, list(model.h_int), shape)
mf_new = model.mfield(mf, nk=nk)
mf_params_new = tb_to_rparams(mf_new)
@@ -65,11 +65,11 @@ def solver(
:
Mean-field correction solution in the tight-binding dictionary format.
"""
- shape = model._size
+ shape = model._ndof
mf_params = tb_to_rparams(mf_guess)
f = partial(cost, model=model, nk=nk)
result = rparams_to_tb(
optimizer(f, mf_params, **optimizer_kwargs), list(model.h_int), shape
)
fermi = calculate_fermi_energy(add_tb(model.h_0, result), model.filling, nk=nk)
- return add_tb(result, {model._local_key: -fermi * np.eye(model._size)})
+ return add_tb(result, {model._local_key: -fermi * np.eye(model._ndof)})
--
GitLab