From 08af692835470c6683d725a41fbf10667dfeff50 Mon Sep 17 00:00:00 2001
From: tsukumi <tsukumijima@users.noreply.github.com>
Date: Mon, 23 Sep 2024 15:20:52 +0900
Subject: [PATCH] Fix: PyTorch inference test

---
 tests/test_main.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/tests/test_main.py b/tests/test_main.py
index 0c529b8..5126232 100644
--- a/tests/test_main.py
+++ b/tests/test_main.py
@@ -47,8 +47,9 @@ def synthesize(
 
                 # ロードされた InferenceSession の ExecutionProvider が一致するか確認
                 # 一致しない場合、指定された ExecutionProvider で推論できない状態
-                assert model.onnx_session is not None
-                assert model.onnx_session.get_providers()[0] == onnx_providers[0][0]
+                if inference_type == "onnx":
+                    assert model.onnx_session is not None
+                    assert model.onnx_session.get_providers()[0] == onnx_providers[0][0]
 
                 # すべてのスタイルに対して音声合成を実行
                 for style in model_info.styles: