Add tiny test for ToPickle

[mrocklin]

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

[github-actions]

Unit Test Results

       17 files  +       6         17 suites  +6   9h 5m 22s ⏱️ + 3h 24m 26s
  2 703 tests +     28    2 621 ✔️ +     30       81 💤  -     2  1 ❌ ±0 
22 967 runs  +8 476  21 740 ✔️ +8 024  1 226 💤 +452  1 ❌ ±0 

For more details on these failures, see this check.

Results for commit 2ac24e5. ± Comparison against base commit ed48736.

♻️ This comment has been updated with latest results.

[madsbk]

LGTM but I suggest that the test is moved to test_to_pickle.py.

[mrocklin]

Planning to merge when CI finishes.

[mrocklin]

Also, it looks like ToPickle fails with inproc comms.

[mrocklin]

Maybe we need this?

diff --git a/distributed/protocol/serialize.py b/distributed/protocol/serialize.py
index b4daf5bd..48091cc5 100644
--- a/distributed/protocol/serialize.py
+++ b/distributed/protocol/serialize.py
@@ -611,8 +611,12 @@ def nested_deserialize(x):
                     x[k] = replace_inner(v)
                 elif typ is Serialize:
                     x[k] = v.data
+                elif typ is ToPickle:
+                    x[k] = v.data
                 elif typ is Serialized:
                     x[k] = deserialize(v.header, v.frames)
+                elif typ is Pickled:
+                    x[k] = pickle.loads(v.header, buffers=v.frames)
 
         elif type(x) is list:
             x = list(x)
@@ -622,8 +626,12 @@ def nested_deserialize(x):
                     x[k] = replace_inner(v)
                 elif typ is Serialize:
                     x[k] = v.data
+                elif typ is ToPickle:
+                    x[k] = v.data
                 elif typ is Serialized:
                     x[k] = deserialize(v.header, v.frames)
+                elif typ is Pickled:
+                    x[k] = pickle.loads(v.header, buffers=v.frames)
 
         return x
 

[madsbk]

#6044 should fix the nproc issue.