Return results in restart_workers

[milesgranger]

Small extension to #7154 to return results from Client.restart_workers

• Tests added / passed
• Passes pre-commit run --all-files

Wasn't completely sure if testing an exception raised during restart ought to be handled here, but prior conversations suggest it can be ignored or otherwise logged as is.

In [3]: client.restart_workers(['tcp://127.0.0.1:37623'])
Out[3]: {'tcp://127.0.0.1:37623': 'OK'}

[GPUtester]

Can one of the admins verify this patch?

Admins can comment ok to test to allow this one PR to run or add to allowlist to allow all future PRs from the same author to run.

[github-actions]

Unit Test Results

See test report for an extended history of previous test failures. This is useful for diagnosing flaky tests.

       24 files  +       1         24 suites  +1   11h 20m 32s ⏱️ + 9m 8s
  3 489 tests ±       0    3 380 ✔️ +       1     106 💤 ±  0  3 ❌  - 1 
40 995 runs  +1 291  39 130 ✔️ +1 263  1 862 💤 +29  3 ❌  - 1 

For more details on these failures, see this check.

Results for commit 1cde09b. ± Comparison against base commit 8585f98.

♻️ This comment has been updated with latest results.

[fjetter]

There appears to be at least one related failure in distributed.tests.test_client:: test_restart_workers

Most of the failing tests are fixed by #7595. If you merge main/rebase this should look much better

[jrbourbeau]

add to allowlist

[jrbourbeau]

Thanks @milesgranger! This is in

Also, I noticed this is your first code contribution to this repository. Welcome!

[fjetter]

Sorry to "reopen" this. I believe we should add tests for a couple of cases with failing worker restarts. This is where this feature actually comes in handy.

[jrbourbeau]

Makes sense -- @milesgranger would you mind adding some failing worker restarts test cases?