diff --git a/scripts/representative.py b/scripts/representative.py index c940686b2126121b33d12a858026338aff737706..a670ad6e5fff761b1e2675e44b72839b2bef313d 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()