Fix aimet-onnx test dependency collision (#3830)

Signed-off-by: Kyunggeun Lee <[email protected]>
quic · Feb 23, 2025 · a95300e · a95300e
1 parent d3e25c5
commit a95300e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/packaging/plugins/local/aimet.py b/packaging/plugins/local/aimet.py
@@ -86,7 +86,6 @@ def optional_dependencies() -> dict[str, list[str]]:
             "beautifulsoup4",
             "deepspeed",
             "matplotlib",
-            "onnxruntime",
             "peft",
             "pylint<3",
             "pytest",
@@ -116,6 +115,8 @@ def optional_dependencies() -> dict[str, list[str]]:
     if aimet_variant not in ("torch-gpu", "torch-cpu"):
         return optional_dependencies
 
+    optional_dependencies["test"].append("onnxruntime")
+
     try:
         import torch
     except ImportError: