rewrite the thing

dask · Jan 9, 2024 · c5f1379 · c5f1379
1 parent ee7c1c5
commit c5f1379
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/distributed/protocol/pickle.py b/distributed/protocol/pickle.py
@@ -67,14 +67,16 @@ def dumps(x, *, buffer_callback=None, protocol=HIGHEST_PROTOCOL):
             buffers.clear()
             pickler.dump(x)
             result = f.getvalue()
-        if (
+
+        if not _always_use_pickle_for(x) and (
             CLOUDPICKLE_GE_20
             and getattr(inspect.getmodule(x), "__name__", None)
             in cloudpickle.list_registry_pickle_by_value()
-        ) or (
-            (len(result) < 1000 and not _always_use_pickle_for(x))
-            # this is pretty expensive so check very last
-            and b"__main__" in result
+            or (
+                len(result) < 1000
+                # Do this very last since it's expensive
+                and b"__main__" in result
+            )
         ):
             buffers.clear()
             result = cloudpickle.dumps(x, **dump_kwargs)