From ef3df2c82444b677379e3225515a904c949e982c Mon Sep 17 00:00:00 2001 From: Weibin Zeng Date: Tue, 23 Feb 2021 21:32:48 +0800 Subject: [PATCH] Put config to handle in get_handle (#159) --- python/graphscope/learning/graph.py | 1 + 1 file changed, 1 insertion(+) diff --git a/python/graphscope/learning/graph.py b/python/graphscope/learning/graph.py index 974e5083438d..e46907145b25 100644 --- a/python/graphscope/learning/graph.py +++ b/python/graphscope/learning/graph.py @@ -202,6 +202,7 @@ def selected_property_schema(attr_types, attributes): def get_handle(self, worker_count=1): """Return a base64-encoded handle for distributed training.""" handle_copy = self.handle.copy() + handle_copy["config"] = self.config handle_copy["client_count"] = worker_count return base64.b64encode(json.dumps(handle_copy).encode("utf-8")).decode("utf-8")