From 49ab4727d3664aae8c1d11247b23f50849678d91 Mon Sep 17 00:00:00 2001
From: Kostas Vilkelis <kostasvilkelis@gmail.com>
Date: Mon, 6 May 2024 23:58:35 +0200
Subject: [PATCH] use Tuple for multiple outputs
---
pymf/mf.py | 7 +++++--
1 file changed, 5 insertions(+), 2 deletions(-)
diff --git a/pymf/mf.py b/pymf/mf.py
index 5f2d82c..dd21ccd 100644
--- a/pymf/mf.py
+++ b/pymf/mf.py
@@ -1,5 +1,6 @@
import numpy as np
from scipy.fftpack import ifftn
+from typing import Tuple
from pymf.tb.tb import add_tb, tb_type
from pymf.tb.transforms import ifftn_to_tb, tb_to_khamvector
@@ -7,7 +8,7 @@ from pymf.tb.transforms import ifftn_to_tb, tb_to_khamvector
def construct_density_matrix_kgrid(
kham: np.ndarray, filling: float
-) -> (np.ndarray, float):
+) -> Tuple[np.ndarray, float]:
"""Calculate density matrix on a k-space grid.
Parameters
@@ -34,7 +35,9 @@ def construct_density_matrix_kgrid(
return density_matrix_krid, fermi
-def construct_density_matrix(h: tb_type, filling: float, nk: int) -> (tb_type, float):
+def construct_density_matrix(
+ h: tb_type, filling: float, nk: int
+) -> Tuple[tb_type, float]:
"""Compute the density matrix in real-space tight-binding format.
Parameters
--
GitLab