Adding typecast to two variables

Author: adrianoaru-nhs

utils/fit_kit_logged.py

@@ -48,7 +48,7 @@ def process_kit_data(smokescreen_properties: dict) -> list:
     return device_ids
 
 
-def split_fit_kits(kit_id_df, smokescreen_properties: dict) -> pd.DataFrame:
+def split_fit_kits(kit_id_df: pd.DataFrame, smokescreen_properties: dict) -> pd.DataFrame:
     """
     This method splits the dataframe into two, 1 normal and 1 abnormal
     """

utils/oracle/oracle.py

@@ -126,7 +126,7 @@ def delete_all_users_from_approved_users_table(
                 self.disconnect_from_db(conn)
 
     def execute_query(
-        self, query: str, parameters = None
+        self, query: str, parameters: list | None = None
     ) -> pd.DataFrame:  # To use when "select xxxx" (stored procedures)
         """
         This is used to execute any sql queries.