diff --git a/tracex_project/extraction/logic/modules/module_activity_labeler.py b/tracex_project/extraction/logic/modules/module_activity_labeler.py index 7e71461..bc8e97e 100644 --- a/tracex_project/extraction/logic/modules/module_activity_labeler.py +++ b/tracex_project/extraction/logic/modules/module_activity_labeler.py @@ -70,7 +70,9 @@ def __number_patient_journey_sentences(patient_journey_sentences: List[str]) -> @staticmethod def __extract_activities( - patient_journey_numbered: str, condition: Optional[str], number_of_senteces: int + patient_journey_numbered: str, + condition: Optional[str], + number_of_sentences: int, ) -> pd.DataFrame: """ Converts a Patient Journey, where every sentence is numbered, to a DataFrame with the activity labels by @@ -91,7 +93,7 @@ def __extract_activities( " #", expand=True ) except ValueError: - scaling_factor = df.shape[0] / (number_of_senteces - 1) + scaling_factor = df.shape[0] / (number_of_sentences - 1) df["sentence_id"] = int(df.reset_index().index * scaling_factor) return df