Skip to content
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.

Commit 8bb6a03

Browse files
committedMar 9, 2025·
Ignore typing issues for now.
1 parent 5bb50a0 commit 8bb6a03

File tree

2 files changed

+8
-6
lines changed

2 files changed

+8
-6
lines changed
 

‎src/pytask_parallel/execute.py

+5-3
Original file line numberDiff line numberDiff line change
@@ -282,12 +282,14 @@ def _update_carry_over_node(
282282
return x
283283
raise NotImplementedError
284284

285-
structure_carry_over_products = tree_structure(carry_over_products)
286-
structure_produces = tree_structure(task.produces)
285+
structure_carry_over_products = tree_structure(carry_over_products) # type: ignore[arg-type]
286+
structure_produces = tree_structure(task.produces) # type: ignore[arg-type]
287287
# strict must be false when none is leaf.
288288
if structure_produces.is_prefix(structure_carry_over_products, strict=False):
289289
task.produces = tree_map(
290-
_update_carry_over_node, task.produces, carry_over_products
290+
_update_carry_over_node,
291+
task.produces, # type: ignore[arg-type]
292+
carry_over_products, # type: ignore[arg-type]
291293
) # type: ignore[assignment]
292294

293295

‎src/pytask_parallel/wrappers.py

+3-3
Original file line numberDiff line numberDiff line change
@@ -269,7 +269,7 @@ def _save_and_carry_over_product(
269269
node.save(value)
270270
return None
271271

272-
return tree_map_with_path(_save_and_carry_over_product, task.produces)
272+
return tree_map_with_path(_save_and_carry_over_product, task.produces) # type: ignore[arg-type]
273273

274274

275275
def _write_local_files_to_remote(
@@ -281,7 +281,7 @@ def _write_local_files_to_remote(
281281
to be resolved.
282282
283283
"""
284-
return tree_map(lambda x: x.load() if isinstance(x, RemotePathNode) else x, kwargs) # type: ignore[return-value]
284+
return tree_map(lambda x: x.load() if isinstance(x, RemotePathNode) else x, kwargs) # type: ignore[arg-type, return-value]
285285

286286

287287
def _delete_local_files_on_remote(kwargs: dict[str, PyTree[Any]]) -> None:
@@ -298,4 +298,4 @@ def _delete(potential_node: Any) -> None:
298298
os.close(potential_node.fd)
299299
Path(potential_node.remote_path).unlink(missing_ok=True)
300300

301-
tree_map(_delete, kwargs)
301+
tree_map(_delete, kwargs) # type: ignore[arg-type]

0 commit comments

Comments
 (0)
Please sign in to comment.