Skip to content

Commit

Permalink
test (ex/llm): Removed onnx dependency when collecting tests.
Browse files Browse the repository at this point in the history
  • Loading branch information
nickfraser committed Aug 23, 2024
1 parent b7c4808 commit d4509c6
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tests/brevitas_examples/test_llm.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,6 @@
import shutil

import numpy as np
import onnx
import pytest
import torch

Expand Down Expand Up @@ -323,6 +322,7 @@ def onnx_export_args(default_run_args, request):
@pytest.mark.llm
@jit_disabled_for_export()
def test_small_models_onnx_export(caplog, onnx_export_args):
import onnx
caplog.set_level(logging.INFO)
args = onnx_export_args
float_ppl, quant_ppl, model = main(args)
Expand Down

0 comments on commit d4509c6

Please sign in to comment.