From 504e04cc8d68348e74970d6e7a01d70e25c16d21 Mon Sep 17 00:00:00 2001
From: Hugo Gillet <hugo.gillet@stud.unibas.ch>
Date: Fri, 2 Dec 2022 11:07:01 +0000
Subject: [PATCH] Update representative.py

---
 scripts/representative.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/scripts/representative.py b/scripts/representative.py
index c940686..a670ad6 100644
--- a/scripts/representative.py
+++ b/scripts/representative.py
@@ -11,7 +11,7 @@ support level for each gene of the input
 
 
 
-def import_gtfSelection_to_df(gtf_modified_file: str):
+def import_gtfSelection_to_df(gtf_modified_file: str) -> pd.DataFrame:
     """Import intermediate file from gtf and create a df
 
         Args:
@@ -41,7 +41,7 @@ names = ["Gene_mixed", "Transcript", "Support_level", "Na1", "Na2"] )
 
 
 
-def representative_transcripts_inDict(df_gtfSelection: str) -> pd.DataFrame:
+def representative_transcripts_inDict(df_gtfSelection: pd.DataFrame) -> pd.DataFrame:
     """Return a dict containing for each gene transcripts 
         with highest confidence level
 
@@ -70,7 +70,7 @@ def representative_transcripts_inDict(df_gtfSelection: str) -> pd.DataFrame:
 
 
 
-def find_repr_by_SupportLevel(intermediate_file:str): 
+def find_repr_by_SupportLevel(intermediate_file : str) -> dict[str,str]: 
     """Combine functions import_gtfSelection_to_df() 
         and representative_transcripts_inDict()
 
-- 
GitLab