3.5.1 (#985)

* Fix type for device id * Fix type for device id * Fix order * Remove comma * Replace Generator typing * Replace Generator typing * Conditional kill myself (#984) * Introduce conditional remove mask * fix --------- Co-authored-by: harisreedhar <h4harisreedhar.s.s@gmail.com> --------- Co-authored-by: harisreedhar <h4harisreedhar.s.s@gmail.com>
2025-11-19 17:39:42 +01:00
parent 81c5e85dea
commit 420d738a6b
16 changed files with 45 additions and 41 deletions
--- a/tests/test_execution.py
+++ b/tests/test_execution.py
@@ -15,10 +15,10 @@ def test_create_inference_session_providers() -> None:
 	[
 		('CUDAExecutionProvider',
 		{
-			'device_id': '1',
+			'device_id': 1,
 			'cudnn_conv_algo_search': 'EXHAUSTIVE'
 		}),
 		'CPUExecutionProvider'
 	]

-	assert create_inference_session_providers('1', [ 'cpu', 'cuda' ]) == inference_session_providers
+	assert create_inference_session_providers(1, [ 'cpu', 'cuda' ]) == inference_session_providers