Add missing import

tharapalanivel · tharapalanivel · commit 5731b5ff3ef9 · 2025-01-07T11:46:34.000-08:00
Signed-off-by: Thara Palanivel &lt;130496890+tharapalanivel@users.noreply.github.com&gt;
diff --git a/tests/test_run_quant.py b/tests/test_run_quant.py
@@ -21,6 +21,7 @@
 
 # Third Party
 import pytest
+import torch
 
 # Local
 from fms_mo.run_quant import get_parser, parse_arguments, quantize
