We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent da2e722 commit 01d25ceCopy full SHA for 01d25ce
_unittests/ut_torch_export_patches/test_patch_module.py
@@ -5,7 +5,7 @@
5
import numpy as np
6
from scipy.spatial.distance import cdist
7
import torch
8
-from onnx_diagnostic.ext_test_case import ExtTestCase, hide_stdout, has_torch
+from onnx_diagnostic.ext_test_case import ExtTestCase, hide_stdout, has_torch, requires_torch
9
from onnx_diagnostic.torch_export_patches import torch_export_patches, torch_export_rewrite
10
from onnx_diagnostic.torch_export_patches.patch_module import (
11
transform_method,
0 commit comments