Update news_category_prediction.py

news_category_prediction.py

@@ -64,7 +64,7 @@ def predict_news_category(old_news: pd.DataFrame, new_news: pd.DataFrame, interp
         if 'category' not in [*old_news.columns]:
             logger.warning('No prior predictions found in old news')
             if not cols_check([*new_news.columns], [*old_news.columns]):
-                raise Exeption("New and old cols don't match")
+                raise Exception("New and old cols don't match")
             final_df = pd.concat([old_news, new_news], axis=0, ignore_index=True)
             final_df.drop_duplicates(subset='url', keep='first', inplace=True)
             headlines = [*final_df['title']].copy()
@@ -74,7 +74,7 @@ def predict_news_category(old_news: pd.DataFrame, new_news: pd.DataFrame, interp
         else:
             logger.warning('Prior predictions found in old news')
             if not cols_check([*new_news.columns], [*old_news.columns][:-2]):
-                raise Exeption("New and old cols don't match")
+                raise Exception("New and old cols don't match")
             old_urls = [*old_news['url']]
             new_news = new_news.loc[new_news['url'].isin(old_urls) == False, :]
             headlines = [*new_news['title']].copy()