test: Test with singer-sdk @ main

MeltanoLabs · Jul 25, 2024 · e6d8190 · e6d8190
1 parent 03c428e
commit e6d8190
Show file tree

Hide file tree

Showing 3 changed files with 23 additions and 20 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -16,7 +16,7 @@ packages = [
 [tool.poetry.dependencies]
 python = ">=3.8"
 cryptography = ">=40,<43"
-singer-sdk = "~=0.38.0"
+singer-sdk = { git = "https://github.com/meltano/sdk.git" }
 snowflake-sqlalchemy = "~=1.6.1"
 snowflake-connector-python = { version = "<4.0.0", extras = ["secure-local-storage"] }
 sqlalchemy = "<2"
@@ -25,7 +25,7 @@ sqlalchemy = "<2"
 coverage = ">=7.2.7"
 pytest = ">=7.4.3"
 pytest-xdist = ">=3.3.1"
-singer-sdk = { version="~=0.38.0", extras = ["testing"] }
+singer-sdk = { git = "https://github.com/meltano/sdk.git", extras = ["testing"] }
 
 [tool.ruff]
 line-length = 120

diff --git a/target_snowflake/sinks.py b/target_snowflake/sinks.py
@@ -21,15 +21,15 @@
 from target_snowflake.connector import SnowflakeConnector
 
 if t.TYPE_CHECKING:
-    from singer_sdk import PluginBase, SQLConnector
+    from singer_sdk import PluginBase
 
 DEFAULT_BATCH_CONFIG = {
     "encoding": {"format": "jsonl", "compression": "gzip"},
     "storage": {"root": "file://"},
 }
 
 
-class SnowflakeSink(SQLSink):
+class SnowflakeSink(SQLSink[SnowflakeConnector]):
     """Snowflake target sink class."""
 
     connector_class = SnowflakeConnector
@@ -40,7 +40,7 @@ def __init__(
         stream_name: str,
         schema: dict,
         key_properties: list[str] | None,
-        connector: SQLConnector | None = None,
+        connector: SnowflakeConnector | None = None,
     ) -> None:
         """Initialize Snowflake Sink."""
         self.target = target