ut

xadupre · xadupre · commit f3e65154fd16 · 2025-04-25T15:29:05.000+02:00
diff --git a/_unittests/ut_helpers/test_cache_helper.py b/_unittests/ut_helpers/test_cache_helper.py
@@ -145,8 +145,8 @@ def test_make_mamba_cache(self):
         )
         text = self.string_type(cache, with_shape=True)
         self.assertEqual(
-            "MambaCache(conv_states=#3[T10s4x4x4,T10s4x4x4,T10s4x4x4], "
-            "ssm_states=#3[T10s4x4x4,T10s4x4x4,T10s4x4x4])",
+            "MambaCache(conv_states=#3[T1s4x4x4,T1s4x4x4,T1s4x4x4], "
+            "ssm_states=#3[T1s4x4x4,T1s4x4x4,T1s4x4x4])",
             text,
         )
 

Original file line number	Diff line number	Diff line change
`@@ -145,8 +145,8 @@ def test_make_mamba_cache(self):`
`145`	`145`	`)`
`146`	`146`	`text = self.string_type(cache, with_shape=True)`
`147`	`147`	`self.assertEqual(`
`148`		`- "MambaCache(conv_states=#3[T10s4x4x4,T10s4x4x4,T10s4x4x4], "`
`149`		`- "ssm_states=#3[T10s4x4x4,T10s4x4x4,T10s4x4x4])",`
	`148`	`+ "MambaCache(conv_states=#3[T1s4x4x4,T1s4x4x4,T1s4x4x4], "`
	`149`	`+ "ssm_states=#3[T1s4x4x4,T1s4x4x4,T1s4x4x4])",`
`150`	`150`	`text,`
`151`	`151`	`)`
`152`	`152`