corrects SB test t bug

atong01 · Nov 23, 2023 · 42273f3 · 42273f3
1 parent d3fe6f2
commit 42273f3
Showing 1 changed file with 2 additions and 2 deletions.
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):