public/immich
1
0
Fork 0
mirror of https://github.com/immich-app/immich.git synced 2025-11-01 02:17:43 +09:00
Code Issues Actions 7 Packages Projects Releases Wiki Activity

fix: test_sets_default_sess_options fails if compiling with globally enabled cuda (#17516)

Browse Source 
Coming from nixos, this test fails when cuda is enabled. https://github.com/NixOS/nixpkgs/pull/382896

Solution as proposed by @mertalev
This commit is contained in:
Mitchell Pleune
2025-04-10 13:06:33 -04:00
committed by GitHub
parent 7fcab4b251
commit 9fd2c5220d
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
machine-learning/test_main.py
View File

@@ -278,8 +278,8 @@ class TestOrtSession:
assert session.provider_options == [] assert session.provider_options == []
def test_sets_default_sess_options(self) -> None: def test_sets_default_sess_options_if_cpu(self) -> None:
session = OrtSession("ViT-B-32__openai") session = OrtSession("ViT-B-32__openai", providers=["CPUExecutionProvider"])
assert session.sess_options.execution_mode == ort.ExecutionMode.ORT_SEQUENTIAL assert session.sess_options.execution_mode == ort.ExecutionMode.ORT_SEQUENTIAL
assert session.sess_options.inter_op_num_threads == 1 assert session.sess_options.inter_op_num_threads == 1