fixed pylint further

DSProjects2024 · Mar 12, 2024 · 56cb912 · 56cb912
1 parent aae6462
commit 56cb912
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/thronetalk-game-of-thrones-summarizer/utils/visualization_generator.py b/thronetalk-game-of-thrones-summarizer/utils/visualization_generator.py
@@ -131,7 +131,7 @@ def pre_process_data_for_character_per_episode(self, character: str) -> list[str
                         dialogue_string += dialogue[1]
                     char_episode_wise_arr.append(dialogue_string)
                 break
-            for j in range(1, 11):          
+            for j in range(1, 11):
                 if i == self.season_from and j >= self.episode_from:
                     episode_mask_df = season_mask_df[season_mask_df['Episode'] == 'e' + str(j)]
                     dialogue_string = ''
@@ -261,7 +261,7 @@ def sentiment_analysis_visualization(self, char_arr: list[str]) -> pd.DataFrame:
                 if i == self.season_from and j >= self.episode_from:
                     episode_wise_desc.append("S"+str(i)+":E"+str(j))
                 elif i == self.season_to and j <= self.episode_to:
-                    episode_wise_desc.append("S"+str(i)+":E"+str(j))           
+                    episode_wise_desc.append("S"+str(i)+":E"+str(j))
                 elif self.season_from < i < self.season_to:
                     episode_wise_desc.append("S"+str(i)+":E"+str(j))
         sentiment_arr.append(episode_wise_desc)