Merge pull request #65 from Demea9000/57-fixa-pandas-warning

textclassifier/TextClassifier.py

@@ -8,6 +8,10 @@ from datetime import date
 import os
 from dotenv import find_dotenv, load_dotenv
 import pandas as pd
+import warnings
+from pandas.core.common import SettingWithCopyWarning
+
+warnings.simplefilter(action="ignore", category=SettingWithCopyWarning)
 
 # Set one directory up into ROOT_PATH
 ROOT_PATH = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
@@ -50,7 +54,7 @@ class TextClassifier:
                     raise Exception("Could not get enough tweets. Please try again. Perhaps try different time range.")
                 continue
         # Make id as type int64
-        self.df['id'] = self.df['id'].copy().astype(int)
+        self.df.loc[:, 'id'] = self.df.id.copy().apply(lambda x: int(x))
         # self.api_key = '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'
         openai.api_key = OPENAI_AUTHTOKEN