Skip to content

Commit

Permalink
Put config to handle in get_handle (#159)
Browse files Browse the repository at this point in the history
  • Loading branch information
acezen authored Feb 23, 2021
1 parent 954f8a6 commit ef3df2c
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions python/graphscope/learning/graph.py
Original file line number Diff line number Diff line change
Expand Up @@ -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")

Expand Down

0 comments on commit ef3df2c

Please sign in to comment.