missing files from previous commit

Author: nickzoic

countess/plugins/csv.py

@@ -172,7 +172,7 @@ def execute(
             table = source
 
         def _write(fh):
-            for num, record_batch in enumerate(table.record_batch()):
+            for num, record_batch in enumerate(table.to_arrow_reader()):
                 write_options = pyarrow.csv.WriteOptions(
                     include_header=self.header.value and num == 0,
                     delimiter=self.delimiter.value,

countess/utils/duckdb.py

@@ -4,7 +4,7 @@
 from typing import Iterable, Optional, Union
 
 from duckdb import DuckDBPyConnection, DuckDBPyRelation
-from duckdb.typing import DuckDBPyType
+from duckdb.sqltypes import DuckDBPyType
 
 logger = logging.getLogger(__name__)