Skip to content

Commit

Permalink
[pre-commit.ci] auto fixes from pre-commit.com hooks
Browse files Browse the repository at this point in the history
for more information, see https://pre-commit.ci
  • Loading branch information
pre-commit-ci[bot] committed Aug 3, 2023
1 parent 0d1993d commit dc9c873
Showing 1 changed file with 6 additions and 5 deletions.
11 changes: 6 additions & 5 deletions samples/sample_tap_duckdb/tap_duckdb/tap.py
Original file line number Diff line number Diff line change
@@ -1,17 +1,20 @@
from __future__ import annotations

import typing as t
from textwrap import dedent

import sqlalchemy as sa

from singer_sdk import SQLConnector, SQLTap
from singer_sdk import typing as th
from singer_sdk.helpers._batch import BaseBatchFileEncoding, BatchConfig
from singer_sdk.streams import SQLStream

if t.TYPE_CHECKING:
from singer_sdk.helpers._batch import BatchConfig


class DuckDBBatchError(Exception):
"""DuckDB batch error."""
pass


class DuckDBConnector(SQLConnector):
Expand All @@ -23,9 +26,7 @@ def get_sqlalchemy_url(self, config) -> str:
class DuckDBStream(SQLStream):
connector_class = DuckDBConnector

def get_batches(self, batch_config: BatchConfig, context = None):
root = batch_config.storage.root
prefix = batch_config.storage.prefix or ""
def get_batches(self, batch_config: BatchConfig, context=None):
filepath = "out/out.ndjson"
table_name = self.fully_qualified_name
files = []
Expand Down

0 comments on commit dc9c873

Please sign in to comment.