Skip to content

Commit 22447b4

Browse files
committed
Fix test guards
Change-Id: Icb8f0fd19645f0c0b2730e8db0d2d2b41a34e2c1
1 parent 7b81eec commit 22447b4

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

tests/python/contrib/test_ethosu/cascader/test_graph.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,10 @@
1414
# KIND, either express or implied. See the License for the
1515
# specific language governing permissions and limitations
1616
# under the License.
17+
import pytest
18+
19+
pytest.importorskip("ethosu.vela")
20+
1721
from tvm.contrib.ethosu.cascader import (
1822
StripeConfig,
1923
Propagator,
@@ -23,8 +27,6 @@
2327
CascaderGraph,
2428
)
2529

26-
import pytest
27-
2830

2931
def test_tensor():
3032
shape = [1, 2, 3]

0 commit comments

Comments
 (0)