nv_ingest.framework.orchestration.ray.stages.meta package#

Submodules#

nv_ingest.framework.orchestration.ray.stages.meta.ray_actor_edge_base module#

class nv_ingest.framework.orchestration.ray.stages.meta.ray_actor_edge_base.RayActorEdge(
max_size: int,
multi_reader: bool = False,
multi_writer: bool = False,
)[source]#

Bases: ABC

Abstract base class for a Ray actor edge used in a RayPipeline.

Parameters:

  • max_size (int) – The maximum size of the edge’s internal queue.

  • multi_reader (bool) – Whether the edge supports multiple concurrent readers.

  • multi_writer (bool) – Whether the edge supports multiple concurrent writers.

abstract get_stats() → Dict[str, int][source]#

Get current statistics for the edge.

Returns:

A dictionary containing statistics (e.g. write_count, read_count, queue_full_count, current_size).

Return type:

Dict[str, int]

abstract read() → Any[source]#

Read an item from the edge.

Returns:

The next item in the edge.

Return type:

Any

abstract write(item: Any) → bool[source]#

Write an item into the edge.

Parameters:

item (Any) – The item to enqueue.

Returns:

True if the item was enqueued successfully.

Return type:

bool

nv_ingest.framework.orchestration.ray.stages.meta.ray_actor_sink_stage_base module#

nv_ingest.framework.orchestration.ray.stages.meta.ray_actor_source_stage_base module#

nv_ingest.framework.orchestration.ray.stages.meta.ray_actor_stage_base module#

nv_ingest.framework.orchestration.ray.stages.meta.udf_parallel_helper module#

Module contents#