Function build_setup_hook_export_entry

python/ray/_private/function_manager.py:59–86

Compute the exported payload and GCS key for a setup hook callable. Args: setup_func: The setup hook function to export. job_id: The job ID to export the setup hook for. Returns: A tuple of (pickled_function, function_id, key).

(
    setup_func: Callable, job_id: JobID
)

def build_setup_hook_export_entry(
    setup_func: Callable, job_id: JobID
) -> tuple[bytes, bytes, bytes]:
    """Compute the exported payload and GCS key for a setup hook callable.

    Args:
        setup_func: The setup hook function to export.
        job_id: The job ID to export the setup hook for.

    Returns:
        A tuple of (pickled_function, function_id, key).
    """
    pickled_function = pickle_dumps(
        setup_func,
        "Cannot serialize the worker_process_setup_hook " f"{setup_func.__name__}",
    )
    function_to_run_id = hashlib.shake_128(pickled_function).digest(
        ray_constants.ID_SIZE
    )
    key = make_function_table_key(
        # This value should match with gcs_function_manager.h.
        # Otherwise, it won't be GC'ed.
        WORKER_PROCESS_SETUP_HOOK_KEY_NAME_GCS.encode(),
        # b"FunctionsToRun",
        job_id,
        function_to_run_id,
    )
    return pickled_function, function_to_run_id, key


class FunctionActorManager: