Fix types

reservoir-data · Apr 26, 2024 · af9f31f · af9f31f
1 parent 66a6a5b
commit af9f31f
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/tap_polarsh/streams.py b/tap_polarsh/streams.py
@@ -21,7 +21,7 @@ def generate_child_contexts(
         self,
         record: dict[str, t.Any],
         context: dict[str, t.Any] | None,  # noqa: ARG002
-    ) -> t.Iterable[dict | None]:
+    ) -> t.Iterable[dict[str, t.Any] | None]:
         """Generate child contexts."""
         yield {
             "organization_id": record["id"],
@@ -61,6 +61,6 @@ def get_url_params(
         """Get URL query parameters."""
         return {
             **super().get_url_params(context, next_page_token),
-            "platform": context["platform"],
-            "organization_name": context["organization_name"],
+            "platform": context["platform"] if context else None,
+            "organization_name": context["organization_name"] if context else None,
         }
diff --git a/tap_polarsh/tap.py b/tap_polarsh/tap.py
@@ -15,7 +15,7 @@
     from tap_polarsh.client import PolarStream
 
 OPENAPI_URL = "https://api.polar.sh/openapi.json"
-STREAMS: list[PolarStream] = [
+STREAMS: t.Sequence[type[PolarStream]] = [
     streams.Organizations,
     streams.Repositories,
     streams.Issues,
@@ -55,7 +55,7 @@ def discover_streams(self) -> list[Stream]:
         Returns:
             A list of Neon Serverless Postgres streams.
         """
-        streams: list[PolarStream[str]] = []
+        streams: list[PolarStream] = []
         openapi_schema = self.get_openapi_schema()
 
         for stream_type in STREAMS: