tests for rvc3 non quantized

luxonis · Jul 17, 2024 · 62ca8dd · 62ca8dd
1 parent 50b6b69
commit 62ca8dd
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/tests/conftest.py b/tests/conftest.py
@@ -153,7 +153,7 @@ def prepare(
         converted_model_path = (
             converted_model_path_prefix / f"{model_name}.dlc"
         )
-    elif service in ["rvc2", "rvc3"]:
+    elif service in ["rvc2", "rvc3", "rvc3_non_quant"]:
         converted_model_path = (
             converted_model_path_prefix
             / "intermediate_outputs"

diff --git a/tests/test_packages/test_rvc3.py b/tests/test_packages/test_rvc3.py
@@ -19,6 +19,10 @@ def test_resnet18_convert(rvc3_resnet18_onnx_env):
     check_convert(rvc3_resnet18_onnx_env)
 
 
+def test_resnet18_non_quant_convert(rvc3_non_quant_resnet18_onnx_env):
+    check_convert(rvc3_non_quant_resnet18_onnx_env)
+
+
 def test_resnet18_ir_convert(rvc3_resnet18_ir_env):
     check_convert(rvc3_resnet18_ir_env)