schrodinger.seam.viz.render module¶

class schrodinger.seam.viz.render.StageStatusCounts(num_pending, num_running, num_completed, num_failed)¶

Bases: tuple

num_completed¶: Alias for field number 2

num_failed¶: Alias for field number 3

num_pending¶: Alias for field number 0

num_running¶: Alias for field number 1

class schrodinger.seam.viz.render.TransformAggregateStatus¶

Bases: Enum

PENDING = 'Pending'¶

RUNNING = 'Running'¶

SUCCEEDED = 'Succeeded'¶

FAILED = 'Failed'¶

class schrodinger.seam.viz.render.Styler(seam_events: List[SeamEvent])¶

Bases: object

__init__(seam_events: List[SeamEvent])¶

style(transform_id: str, expanded_nodes: List[str]) → str¶

class schrodinger.seam.viz.render.StageStatus¶

Bases: Enum

PENDING = 1¶

RUNNING = 2¶

COMPLETED = 3¶

FAILED = 4¶

class schrodinger.seam.viz.render.StageInfo(status: StageStatus, wall_time_seconds: float = 0.0)¶

Bases: object

status: StageStatus¶

wall_time_seconds: float = 0.0¶

__init__(status: StageStatus, wall_time_seconds: float = 0.0) → None¶

class schrodinger.seam.viz.render.ColorByStatusStyler(seam_events: List[SeamEvent])¶

Bases: _AbstractTransformStatusStyler

Transform status styler that colors the top bar based on the status of the stages that make up the transform.

The top bar is segmented if the transform has multiple stages with different statuses. Each segment is colored based on the status of the stage it represents.

style_top_bar_row(transform_id: str) → str¶

class schrodinger.seam.viz.render.ColorByStageStatusStyler(seam_events: List[SeamEvent])¶

Bases: Styler

__init__(seam_events: List[SeamEvent])¶

style(transform_id: str, expanded_nodes: List[str]) → str¶

class schrodinger.seam.viz.render.ColorByLicenseStyler(seam_events: List[SeamEvent])¶

Bases: Styler

Implements styling based on the licenses required by the transforms.

The style for a transform is determined by the licenses required by the transform and its subtransforms. The style is a striped rectangle with colors based on the licenses required by the transform. The colors are determined by a hash of the license id. If a transform requires multiple licenses, the colors are combined in a striped pattern.

__init__(seam_events: List[SeamEvent])¶

style(transform_id: str, expanded_nodes) → str¶

class schrodinger.seam.viz.render.ColorByTransformCPUTimeStyler(seam_events: List[SeamEvent])¶

Bases: Styler

COLOR_GRADIENT = ['#B3E2BD', '#FFFFB3', '#FB8071']¶

__init__(seam_events: List[SeamEvent])¶

style(transform_id: str, expanded_nodes: List[str]) → str¶

class schrodinger.seam.viz.render.PipelineRenderer(pipeline_proto, options, expanded_nodes: Optional[List[str]] = None)¶

Bases: PipelineRenderer

A PipelineRenderer that has a few extensions from the base beam PipelineRender:

It can render a pipeline with a subset of the nodes expanded.

It can render a pipeline with custom node styles.

The rendered graph is clickable and will call a javascript function
when a node is clicked.

Intended mostly for use with the seam watcher.

__init__(pipeline_proto, options, expanded_nodes: Optional[List[str]] = None)¶

to_dot_iter()¶

setStyler(styler: Styler)¶: Sets a callback that will be called to get the style for a transform. The callback should take a transform id and return a string that will be used as the style for that transform.

style(transform_id)¶

transform_attributes(transform_id)¶

to_dot()¶

classmethod userPipelineFromSeamEvents(seam_events: ~typing.List[~schrodinger.seam.runners.events.SeamEvent], expanded_nodes: ~typing.Optional[~typing.List[str]] = None, styler_cls: ~schrodinger.seam.viz.render.Styler = <class 'schrodinger.seam.viz.render.ColorByStatusStyler'>) → str¶

Renders a pipeline from a list of seam events and returns the dot string. The pipeline will be rendered with the user labels as the nodes.

The rendered graph is clickable and will call a javascript function when a node is clicked.

classmethod stagePipelineFromSeamEvents(seam_events: List[SeamEvent])¶: Renders a pipeline from a list of seam events. The pipeline will be rendered with the stage names as the nodes. The nodes will be colored based on the status of the stage (e.g. running, completed). The rendered graph