Simplify Cython ID types and fix bug of ActorCheckpointID (#4045)

python/ray/__init__.py

@@ -61,19 +61,32 @@ from ray._raylet import (
     ObjectID,
     TaskID,
     UniqueID,
-    _ID_TYPES,
 )  # noqa: E402
 
 _config = _Config()
 
 from ray.profiling import profile  # noqa: E402
-from ray.worker import (error_info, init, connect, disconnect, get, put, wait,
-                        remote, get_gpu_ids, get_resource_ids, get_webui_url,
-                        register_custom_serializer, shutdown,
-                        is_initialized)  # noqa: E402
-from ray.worker import (SCRIPT_MODE, WORKER_MODE, LOCAL_MODE,
-                        PYTHON_MODE)  # noqa: E402
-from ray.worker import global_state  # noqa: E402
+from ray.worker import (
+    LOCAL_MODE,
+    PYTHON_MODE,
+    SCRIPT_MODE,
+    WORKER_MODE,
+    connect,
+    disconnect,
+    error_info,
+    get,
+    get_gpu_ids,
+    get_resource_ids,
+    get_webui_url,
+    global_state,
+    init,
+    is_initialized,
+    put,
+    register_custom_serializer,
+    remote,
+    shutdown,
+    wait,
+)  # noqa: E402
 import ray.internal  # noqa: E402
 # We import ray.actor because some code is run in actor.py which initializes
 # some functions in the worker.
@@ -84,14 +97,34 @@ from ray.actor import method  # noqa: E402
 __version__ = "0.7.0.dev0"
 
 __all__ = [
-    "error_info", "init", "connect", "disconnect", "get", "put", "wait",
-    "remote", "profile", "actor", "method", "get_gpu_ids", "get_resource_ids",
-    "get_webui_url", "register_custom_serializer", "shutdown",
-    "is_initialized", "SCRIPT_MODE", "WORKER_MODE", "LOCAL_MODE",
-    "PYTHON_MODE", "global_state", "_config", "__version__", "internal",
-    "_ID_TYPES"
+    "LOCAL_MODE",
+    "PYTHON_MODE",
+    "SCRIPT_MODE",
+    "WORKER_MODE",
+    "__version__",
+    "_config",
+    "actor",
+    "connect",
+    "disconnect",
+    "error_info",
+    "get",
+    "get_gpu_ids",
+    "get_resource_ids",
+    "get_webui_url",
+    "global_state",
+    "init",
+    "internal",
+    "is_initialized",
+    "method",
+    "profile",
+    "put",
+    "register_custom_serializer",
+    "remote",
+    "shutdown",
+    "wait",
 ]
 
+# ID types
 __all__ += [
     "ActorCheckpointID",
     "ActorClassID",