Skip to content

Commit 226c378

Browse files
committed
fix: revert some mypy changes
1 parent e17c587 commit 226c378

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

awswrangler/_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -662,7 +662,7 @@ def chunkify(
662662
if not lst:
663663
return []
664664
n: int = num_chunks if max_length is None else int(math.ceil(float(len(lst)) / float(max_length)))
665-
np_chunks = np.array_split(lst, n)
665+
np_chunks = np.array_split(lst, n) # type: ignore[arg-type,var-annotated]
666666
return [arr.tolist() for arr in np_chunks if len(arr) > 0]
667667

668668

awswrangler/distributed/ray/datasources/arrow_parquet_datasource.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -334,16 +334,16 @@ def get_read_tasks(self, parallelism: int) -> list[ReadTask]:
334334
)
335335

336336
read_tasks = []
337-
for fragments, paths, metadata in zip(
337+
for fragments, paths, metadata in zip( # type: ignore[var-annotated]
338338
np.array_split(pq_fragments, parallelism),
339339
np.array_split(pq_paths, parallelism),
340-
np.array_split(pq_metadata, parallelism),
340+
np.array_split(pq_metadata, parallelism), # type: ignore[arg-type]
341341
):
342342
if len(fragments) <= 0:
343343
continue
344344

345345
meta = self._meta_provider(
346-
paths,
346+
paths, # type: ignore[arg-type]
347347
self._inferred_schema,
348348
num_fragments=len(fragments),
349349
prefetched_metadata=metadata,

0 commit comments

Comments
 (0)