From 67d92497377b791116a61c35abacc88d8f54e66c Mon Sep 17 00:00:00 2001
From: gitgud5000 <17186026+gitgud5000@users.noreply.github.com>
Date: Fri, 28 Feb 2025 04:32:19 -0400
Subject: [PATCH] Fix: Future instantiation in parallel_calculate_chunks for
 correct client usage

---
 featuretools/computational_backends/calculate_feature_matrix.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/featuretools/computational_backends/calculate_feature_matrix.py b/featuretools/computational_backends/calculate_feature_matrix.py
index d91943d37e..435249b787 100644
--- a/featuretools/computational_backends/calculate_feature_matrix.py
+++ b/featuretools/computational_backends/calculate_feature_matrix.py
@@ -744,7 +744,7 @@ def parallel_calculate_chunks(
         _saved_features = client.scatter(pickled_feats)
         client.replicate([_es, _saved_features])
         num_scattered_workers = len(
-            client.who_has([Future(es_token)]).get(es_token, []),
+            client.who_has([Future(es_token, client=client)]).get(es_token, []),
         )
         num_workers = len(client.scheduler_info()["workers"].values())