schrodinger.seam.testing.utils module

schrodinger.seam.testing.utils.visualize_pipelines(name)

class schrodinger.seam.testing.utils.LicenseHintsCollector

Bases: PipelineVisitor

Visitor to collect license hints from graph nodes.

__init__()

visit_transform(node)

Callback for visiting a transform leaf node in the pipeline DAG.

class schrodinger.seam.testing.utils.ComputeHintsCollector

Bases: PipelineVisitor

Visitor to collect compute hints from graph nodes.

__init__()

visit_transform(node)

Callback for visiting a transform leaf node in the pipeline DAG.

class schrodinger.seam.testing.utils.LocalExecutionHintCollector

Bases: PipelineVisitor

Visitor to collect local host hints from graph nodes.

__init__()

enter_composite_transform(transform_node)

Callback for entering traversal of a composite transform node.

visit_transform(node)

Callback for visiting a transform leaf node in the pipeline DAG.

class schrodinger.seam.testing.utils.ProductsRequirementsCollector

Bases: PipelineVisitor

Visitor to collect products hints from graph nodes.

__init__()

visit_transform(node)

Callback for visiting a transform leaf node in the pipeline DAG.