Skip to content

Commit 99093ca

Browse files
committed
proper skip qeinsum test when condition not met
1 parent 4c5d25f commit 99093ca

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

test/pytest/test_qeinsum.py

+2-4
Original file line numberDiff line numberDiff line change
@@ -3,19 +3,17 @@
33
import keras
44
import numpy as np
55
import pytest
6-
from keras.api.layers import Input
76

87
from hls4ml.converters import convert_from_keras_model
98

10-
if keras.__version__ < '3.0.0':
11-
pytest.skip('Only keras v3 is supported for now', allow_module_level=True)
12-
139
try:
1410
from hgq.layers import QEinsum
1511
from hgq.utils import trace_mode
1612
except ImportError:
1713
pytest.skip('s-quark is not installed', allow_module_level=True)
1814

15+
from keras.api.layers import Input
16+
1917
test_root_path = Path(__file__).parent
2018

2119

0 commit comments

Comments
 (0)