@@ -222,7 +222,7 @@ def test_non_memory_error_reraises_same_object(self):
222222
223223 def test_memory_error_triggers_cleanup_and_returns_retry_result (self ):
224224 """A BFCArena error calls cleanup once and returns the retry result."""
225- err = Exception ("BFCArena failed" )
225+ err = RuntimeError ("BFCArena failed" )
226226 cleanup = Mock ()
227227 retry = Mock (return_value = "retried" )
228228
@@ -236,7 +236,7 @@ def test_memory_error_triggers_cleanup_and_returns_retry_result(self):
236236
237237 def test_cpu_fallback_returns_result_session_provider_tuple (self ):
238238 """CPU fallback returns (result, new_session, provider)."""
239- err = Exception ("BFCArena failed" )
239+ err = RuntimeError ("BFCArena failed" )
240240 session_mock = MagicMock ()
241241 creator = Mock (return_value = (session_mock , "CPUExecutionProvider" ))
242242 retry = Mock (return_value = "r" )
@@ -255,7 +255,7 @@ def test_cpu_fallback_returns_result_session_provider_tuple(self):
255255
256256 def test_retry_failure_propagates_retry_exception (self ):
257257 """If the retry itself fails, that exception propagates."""
258- err = Exception ("Failed to allocate memory for requested buffer" )
258+ err = RuntimeError ("Failed to allocate memory for requested buffer" )
259259 retry_err = RuntimeError ("still failing" )
260260 retry = Mock (side_effect = retry_err )
261261
@@ -266,9 +266,9 @@ def test_retry_failure_propagates_retry_exception(self):
266266
267267 def test_memory_error_without_retry_or_fallback_reraises_original (self ):
268268 """A memory error with no retry and no fallback re-raises the original."""
269- err = Exception ("out of memory" )
269+ err = RuntimeError ("out of memory" )
270270
271- with pytest .raises (Exception ) as excinfo :
271+ with pytest .raises (RuntimeError ) as excinfo :
272272 handle_onnx_memory_error (err , "test context" )
273273
274274 assert excinfo .value is err
0 commit comments