type hints, #190

zqfang · Feb 7, 2023 · 4e29b42 · 4e29b42
1 parent cddfa87
commit 4e29b42
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/gseapy/gsea.py b/gseapy/gsea.py
@@ -127,7 +127,7 @@ def calculate_metric(
         neg: str,
         classes: Dict[str, List[str]],
         ascending: bool,
-    ) -> Tuple[List[int], pd.Series[float]]:
+    ) -> Tuple[List[int], pd.Series]:
         """The main function to rank an expression table. works for 2d array.
 
         :param df:      gene_expression DataFrame.

diff --git a/tests/test_commands.py b/tests/test_commands.py
@@ -104,8 +104,8 @@ def test_ssgsea2(ssGCT, geneGMT):
     tmpdir = TemporaryDirectory(dir="tests")
     ssgsea(ssGCT, geneGMT, tmpdir.name, permutation_num=0)
     tmpdir.cleanup()
-    ssgsea(ssGCT, geneGMT, None, permutation_num=0, correl_norm_type='symrank')
-    ssgsea(ssGCT, geneGMT, None, permutation_num=0, correl_norm_type='zscore')
+    ssgsea(ssGCT, geneGMT, None, permutation_num=0, correl_norm_type="symrank")
+    ssgsea(ssGCT, geneGMT, None, permutation_num=0, correl_norm_type="zscore")
 
 
 def test_enrichr(genelist, geneGMT):