chore: type hints

src/dataset/data_requests.py

@@ -18,7 +18,7 @@ def data_prep() -> pd.DataFrame:
     df = clean_date(df)
     return df
 
-def filter_data(df) -> pd.DataFrame: 
+def filter_data(df:pd.DataFrame) -> pd.DataFrame: 
     """
     Filter the DataFrame based on user-selected ranges for latitude, longitude, and date.
     Args:
@@ -36,7 +36,7 @@ def filter_data(df) -> pd.DataFrame:
     ]
     return df_filtered
 
-def show_specie_author(df): 
+def show_specie_author(df:pd.DataFrame): 
     """
     Display a list of species and their corresponding authors with checkboxes.
     Args:
@@ -51,7 +51,7 @@ def show_specie_author(df):
             label = f"{row['author_email']} ({row['counts']})"
             st.session_state.checkbox_states[key] = st.checkbox(label, key=key)
 
-def show_new_data_view(df) -> pd.DataFrame: 
+def show_new_data_view(df:pd.DataFrame) -> pd.DataFrame: 
     """
     Show the new filtered data view on the UI.
     Filter the dataframe based on the state of the localisation sliders and selected timeframe by the user.

src/dataset/fake_data.py

@@ -1,3 +1,4 @@
+from typing import Tuple
 import pandas as pd
 import random
 from datetime import datetime, timedelta
@@ -5,7 +6,7 @@ from datetime import datetime, timedelta
 from dataset.download import presentation_data_schema
 from whale_viewer import WHALE_CLASSES
 
-def generate_fake_data(df, num_fake) -> pd.DataFrame:
+def generate_fake_data(df:pd.DataFrame, num_fake:int) -> pd.DataFrame:
     """
     Generate fake data for the dataset.
 
@@ -23,13 +24,13 @@ def generate_fake_data(df, num_fake) -> pd.DataFrame:
         '[email protected]', '[email protected]'
     ]
 
-    def random_ocean_coord():
+    def random_ocean_coord() -> Tuple[float, float]:
         """Generate random ocean-friendly coordinates."""
         lat = random.uniform(-60, 60)  # avoid poles
         lon = random.uniform(-180, 180)
         return lat, lon
 
-    def random_date(start_year=2018, end_year=2025):
+    def random_date(start_year:int=2018, end_year:int=2025) -> datetime:
         """Generate a random date."""
         start = datetime(start_year, 1, 1)
         end = datetime(end_year, 1, 1)