From 931e516a08a6e3a7f9f4bf7d7f1fe48055354151 Mon Sep 17 00:00:00 2001 From: Johanna <johanna@zijderveld.de> Date: Mon, 30 Oct 2023 15:34:11 +0100 Subject: [PATCH] renaming and changing _params to outside loop --- codes/utils.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/codes/utils.py b/codes/utils.py index 339c305..8e6dc09 100644 --- a/codes/utils.py +++ b/codes/utils.py @@ -184,14 +184,14 @@ def kgrid_hamiltonian(nk, tb_model, return_ks=False): ks : 1D-array List of k-points over all directions. Only returned if `return_ks=True`. """ - dim = len(next(iter(syst))) + dim = len(next(iter(tb_model))) if dim == 0: if return_ks: - return syst[next(iter(syst))], None + return syst[next(iter(tb_model))], None else: - return syst[next(iter(syst))] + return syst[next(iter(tb_model))] else: - hk = dict2hk(syst) + hk = dict2hk(tb_model) ks = 2 * np.pi * np.linspace(0, 1, nk, endpoint=False) -- GitLab