fix

chdsbd · Apr 25, 2024 · 19b8618 · 19b8618
1 parent c58da79
commit 19b8618
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/web_api/s/squawk.py b/web_api/s/squawk.py
@@ -6,7 +6,6 @@
 import subprocess
 import sys
 from pathlib import Path
-from typing import Optional
 
 SQUAWK_VERSION = "0.5.0"
 APP_LABEL = "web_api"
@@ -21,7 +20,7 @@
 _MIGRATION_REGEX = re.compile(r"^(\d{4,}_\w+)\.py$")
 
 
-def _get_migration_id(filepath: str) -> Optional[str]:
+def _get_migration_id(filepath: str) -> str | None:
     """
     valid migrations:
         0001_initial.py
@@ -39,7 +38,7 @@ def _get_migration_id(filepath: str) -> Optional[str]:
     return match.groups()[0]
 
 
-def _get_migration_ids() -> list[str]:
+def _get_migration_ids() -> list[tuple[str, str | None]]:
     current_branch = subprocess.run(
         ["git", "branch", "--show-current"], capture_output=True, check=True
     )