Skip to content

Commit 2f40b9d

Browse files
simonjayhawkinsproost
authored andcommitted
TYP: some types for pandas/io/formats/csvs.py (pandas-dev#30000)
1 parent c25f641 commit 2f40b9d

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

pandas/io/formats/csvs.py

+4-4
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@
55
import csv as csvlib
66
from io import StringIO
77
import os
8+
from typing import Any, Dict, List
89
import warnings
910
from zipfile import ZipFile
1011

@@ -187,7 +188,7 @@ def save(self):
187188
close = True
188189

189190
try:
190-
writer_kwargs = dict(
191+
writer_kwargs: Dict[str, Any] = dict(
191192
lineterminator=self.line_terminator,
192193
delimiter=self.sep,
193194
quoting=self.quoting,
@@ -198,8 +199,7 @@ def save(self):
198199
if self.encoding == "ascii":
199200
self.writer = csvlib.writer(f, **writer_kwargs)
200201
else:
201-
writer_kwargs["encoding"] = self.encoding
202-
self.writer = UnicodeWriter(f, **writer_kwargs)
202+
self.writer = UnicodeWriter(f, encoding=self.encoding, **writer_kwargs)
203203

204204
self._save()
205205

@@ -233,7 +233,7 @@ def _save_header(self):
233233
cols = self.cols
234234
has_mi_columns = self.has_mi_columns
235235
header = self.header
236-
encoded_labels = []
236+
encoded_labels: List[str] = []
237237

238238
has_aliases = isinstance(header, (tuple, list, np.ndarray, ABCIndexClass))
239239
if not (has_aliases or self.header):

0 commit comments

Comments
 (0)