calling register_custom_serializer require ray to be initialized (#7752)

2026-06-28 01:00:10 +08:00 · 2020-03-27 01:24:06 +08:00
parent bcf963a53b
commit c1b05b720d
1 changed files with 3 additions and 0 deletions
@@ -1391,6 +1391,9 @@ def register_custom_serializer(cls,
        use_dict: Deprecated.
        class_id (str): Unique ID of the class. Autogenerated if None.
    """
+    worker = global_worker
+    worker.check_connected()
+
    if use_pickle:
        raise DeprecationWarning(
            "`use_pickle` is no longer a valid parameter and will be removed "