schrodinger.seam.testing.stu_tests.fixed_scale_transform module

Test that the experimental_fixed_scale_transforms feature works correctly.

This workflow creates a simple pipeline with a custom PTransform that is marked for fixed scaling. The test verifies that the ScaledTransformHint is correctly applied to the marked transforms.

Usage:

$SCHRODINGER/run python3 -m schrodinger.seam.testing.stu_tests.fixed_scale_transform

class schrodinger.seam.testing.stu_tests.fixed_scale_transform.ExpensiveCalculation(label: Optional[str] = None)

Bases: PTransform

A transform that simulates an expensive calculation.

expand(pcoll)

class schrodinger.seam.testing.stu_tests.fixed_scale_transform.CheapCalculation(label: Optional[str] = None)

Bases: PTransform

A transform that simulates a cheap calculation.

expand(pcoll)

schrodinger.seam.testing.stu_tests.fixed_scale_transform.main()