From 3083e95f7ac7cfa2201e72f0536f94afde5fd82a Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Sun, 9 Apr 2023 18:25:30 -0500
Subject: [PATCH] Add DistributedRecv cast in distributed.partition for mypy
1.2
---
pytato/distributed/partition.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/pytato/distributed/partition.py b/pytato/distributed/partition.py
index 1549d97..3da47b9 100644
--- a/pytato/distributed/partition.py
+++ b/pytato/distributed/partition.py
@@ -104,7 +104,7 @@ def _map_distributed_graph_partition_nodes(
parts={
pid: replace(part,
input_name_to_recv_node={
- in_name: map_array(recv)
+ in_name: cast(DistributedRecv, map_array(recv))
for in_name, recv in part.input_name_to_recv_node.items()},
output_name_to_send_node={
out_name: map_send(send)
--
GitLab