corrects SB test t bug

kilianFatras · kilianFatras · commit 42273f35ce84 · 2023-11-23T15:31:21.000-05:00
diff --git 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):

Original file line number	Diff line number	Diff line change
`@@ -23,7 +23,7 @@`
`23`	`23`	`ConditionalFlowMatcher(sigma=0.0),`
`24`	`24`	`ExactOptimalTransportConditionalFlowMatcher(sigma=0.0),`
`25`	`25`	`TargetConditionalFlowMatcher(sigma=0.0),`
`26`		`- SchrodingerBridgeConditionalFlowMatcher(sigma=0.0),`
	`26`	`+ SchrodingerBridgeConditionalFlowMatcher(sigma=0.1),`
`27`	`27`	`VariancePreservingConditionalFlowMatcher(sigma=0.0),`
`28`	`28`	`],`
`29`	`29`	`)`
`@@ -46,7 +46,7 @@ def test_random_Tensor_t(FM):`
`46`	`46`	`"FM",`
`47`	`47`	`[`
`48`	`48`	`ExactOptimalTransportConditionalFlowMatcher(sigma=0.0),`
`49`		`- SchrodingerBridgeConditionalFlowMatcher(sigma=0.0),`
	`49`	`+ SchrodingerBridgeConditionalFlowMatcher(sigma=0.1),`
`50`	`50`	`],`
`51`	`51`	`)`
`52`	`52`	`def test_guided_random_Tensor_t(FM):`