Skip to content

Commit

Permalink
Skip onnx test cases if no onnx (apache#9272)
Browse files Browse the repository at this point in the history
This was missing a guard which meant VS Code errored on test collection.
  • Loading branch information
Mousius authored and ylc committed Jan 7, 2022
1 parent 9eb05a8 commit 0c4d5f2
Showing 1 changed file with 9 additions and 4 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -14,15 +14,20 @@
# KIND, either express or implied. See the License for the
# specific language governing permissions and limitations
# under the License.
import tvm
import os
import sys
from tvm import relay
from tvm.relay import quantize as qtz
import logging

import pytest

pytest.importorskip("onnx")

import onnx

import tvm
from tvm import relay
from tvm.relay import quantize as qtz
import tvm.testing
import mxnet as mx
from test_quantization_accuracy import Config, get_val_data, eval_acc

logging.basicConfig(level=logging.INFO)
Expand Down

0 comments on commit 0c4d5f2

Please sign in to comment.