"src/diffusers/pipelines/kandinsky3/__init__.py" did not exist on "a7ca03aa85f94574f06576d2155b3ec061fe8d63"
Unverified Commit 996a9364 authored by Andrei Ivanov's avatar Andrei Ivanov Committed by GitHub
Browse files

[DistDGL] Fixing bug in `test_dist_objects` (#7179)

parent 3fd6a5d0
......@@ -44,7 +44,6 @@ def run_server(
server_count,
num_clients,
shared_mem,
keep_alive=False,
):
# server_count = num_servers_per_machine
g = dgl.distributed.DistGraphServer(
......@@ -55,7 +54,6 @@ def run_server(
graph_path + "/{}.json".format(graph_name),
disable_shared_mem=not shared_mem,
graph_format=["csc", "coo"],
keep_alive=keep_alive,
)
print("start server", server_id)
g.start()
......@@ -774,7 +772,6 @@ if mode == "server":
server_count=num_servers_per_machine,
num_clients=num_part * num_client_per_machine,
shared_mem=shared_mem,
keep_alive=False,
)
elif mode == "client":
os.environ["DGL_NUM_SERVER"] = str(num_servers_per_machine)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment