Skip to content

Commit 01d25ce

Browse files
committed
fix
1 parent da2e722 commit 01d25ce

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

_unittests/ut_torch_export_patches/test_patch_module.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
import numpy as np
66
from scipy.spatial.distance import cdist
77
import torch
8-
from onnx_diagnostic.ext_test_case import ExtTestCase, hide_stdout, has_torch
8+
from onnx_diagnostic.ext_test_case import ExtTestCase, hide_stdout, has_torch, requires_torch
99
from onnx_diagnostic.torch_export_patches import torch_export_patches, torch_export_rewrite
1010
from onnx_diagnostic.torch_export_patches.patch_module import (
1111
transform_method,

0 commit comments

Comments
 (0)