Move ort import within function (#1975)

neuralmagic · Jan 22, 2024 · 805d2fc · 805d2fc
1 parent 27db537
commit 805d2fc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/sparseml/export/validators.py b/src/sparseml/export/validators.py
@@ -20,7 +20,6 @@
 from typing import Callable, List, Optional, Union
 
 import numpy
-import onnxruntime as ort
 
 from sparseml.export.export_data import InputsNames, LabelNames, OutputsNames
 from sparseml.export.helpers import ONNX_MODEL_NAME
@@ -134,6 +133,7 @@ def validate_correctness(
     :param validation_function: The function that will be used to validate the outputs.
     :return: True if the validation passes, False otherwise.
     """
+    import onnxruntime as ort
 
     sample_inputs_path = os.path.join(target_path, InputsNames.basename.value)
     sample_outputs_path = os.path.join(target_path, OutputsNames.basename.value)