Skip to content

Commit e0ad620

Browse files
authored
Merge pull request #33 from cavusmustafa/unit_test_fix_with_nncf
Fix for unit tests with nncf
2 parents 4eee40d + 7e7711f commit e0ad620

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

backends/openvino/tests/test_runner.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,8 @@
11
import argparse
22
import unittest
33

4+
import nncf.torch
5+
46

57
class OpenvinoTestSuite(unittest.TestSuite):
68

@@ -66,7 +68,8 @@ def parse_arguments():
6668
# Discover all existing op tests in "ops" folder
6769
suite = loader.discover(test_params["test_type"], pattern=test_params["pattern"])
6870
# Start running tests
69-
result = unittest.TextTestRunner().run(suite)
71+
with nncf.torch.disable_patching():
72+
result = unittest.TextTestRunner().run(suite)
7073
if result.wasSuccessful():
7174
print("OpenVINO backend tests completed successfully")
7275
else:

0 commit comments

Comments
 (0)