some mypy fixes

pbp/json_generator/gen_abstract.py

@@ -2,6 +2,8 @@
 # Filename: metadata/generator/gen_abstract.py
 # Description:  Abstract class that captures sound wav metadata
 from datetime import datetime
+from typing import List
+
 import pandas as pd
 from pbp.logging_helper import PbpLogger
 
@@ -12,7 +14,7 @@ def __init__(
         logger: PbpLogger,
         audio_loc: str,
         json_base_dir: str,
-        prefix: [str],
+        prefix: List[str],
         start: datetime,
         end: datetime,
         seconds_per_file: float = 0.0,

pbp/json_generator/gen_iclisten.py

@@ -5,6 +5,8 @@
 import re
 from datetime import timedelta
 from datetime import datetime
+from typing import List
+
 import boto3
 
 import pandas as pd
@@ -27,7 +29,7 @@ def __init__(
         json_base_dir: str,
         start: datetime,
         end: datetime,
-        prefix: [str],
+        prefix: List[str],
         seconds_per_file: float = 300.0,
     ):
         """

pbp/json_generator/gen_nrs.py

@@ -5,6 +5,8 @@
 import re
 from datetime import timedelta, datetime
 import time
+from typing import List
+
 from google.cloud import storage
 
 import pandas as pd
@@ -25,7 +27,7 @@ def __init__(
         json_base_dir: str,
         start: datetime,
         end: datetime,
-        prefix: [str],
+        prefix: List[str],
         seconds_per_file: float = 14400.0,
     ):
         """

pbp/json_generator/gen_soundtrap.py

@@ -2,6 +2,7 @@
 # Filename: json_generator/gen_soundtrap.py
 # Description:  Captures SoundTrap metadata either from a local directory of S3 bucket
 import logging
+from typing import List
 
 import boto3
 import datetime
@@ -34,7 +35,7 @@ def __init__(
         pbp_logger: PbpLogger,
         uri: str,
         json_base_dir: str,
-        prefix: [str],
+        prefix: List[str],
         start: datetime,
         end: datetime,
     ):

pbp/json_generator/utils.py

@@ -1,7 +1,8 @@
+from typing import Tuple
 from urllib.parse import urlparse
 
 
-def parse_s3_or_gcp_url(url) -> (str, str, str):
+def parse_s3_or_gcp_url(url) -> Tuple[str, str, str]:
     """
     Parse the S3, GS of local file url
     :param url:

pyproject.toml

@@ -52,8 +52,10 @@ ignore_missing_imports = true
 exclude = [
     'examples/',
     'pbp/plotting\.py',
-    'pbp/json_generator/',
-    'tests/test_json_generator.py',
+    'pbp/json_generator/metadata.*\.py',
+    'pbp/json_generator/corrector\.py',
+    'pbp/json_generator/gen_soundtrap\.py',
+    'pbp/json_generator/gen_nrs\.py',
 ]
 
 [tool.ruff]