diff --git a/tests/test_time_t.py b/tests/test_time_t.py index 21ad5cc..89d9499 100644 --- a/tests/test_time_t.py +++ b/tests/test_time_t.py @@ -23,7 +23,7 @@ ConditionalFlowMatcher(sigma=0.0), ExactOptimalTransportConditionalFlowMatcher(sigma=0.0), TargetConditionalFlowMatcher(sigma=0.0), - SchrodingerBridgeConditionalFlowMatcher(sigma=0.0), + SchrodingerBridgeConditionalFlowMatcher(sigma=0.1), VariancePreservingConditionalFlowMatcher(sigma=0.0), ], ) @@ -46,7 +46,7 @@ def test_random_Tensor_t(FM): "FM", [ ExactOptimalTransportConditionalFlowMatcher(sigma=0.0), - SchrodingerBridgeConditionalFlowMatcher(sigma=0.0), + SchrodingerBridgeConditionalFlowMatcher(sigma=0.1), ], ) def test_guided_random_Tensor_t(FM):