Skip to content

Commit 227defe

Browse files
Skip training tests on windows (#930)
* Skip training tests on windows * Trigger Tests * Trigger Tests
1 parent 966c8c2 commit 227defe

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

tests/openvino/test_training.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,7 @@
1212
# See the License for the specific language governing permissions and
1313
# limitations under the License.
1414

15+
import os
1516
import random
1617
import re
1718
import shutil
@@ -89,6 +90,10 @@ def initialize_movement_sparsifier_parameters_by_sparsity(
8990
operand.bias_importance.copy_(bias_init_tensor)
9091

9192

93+
def is_windows():
94+
return os.name == "nt"
95+
96+
9297
def is_avx_vnni_supported() -> bool:
9398
return any(re.search("avx.*vnni", flag.lower()) is not None for flag in cpuinfo.get_cpu_info()["flags"])
9499

@@ -614,6 +619,7 @@ def check_ovmodel_reshaping(self, ovmodel: OVModel):
614619
# TODO : can be moved to MODEL_NAMES["swin-window"] after transformers v4.42.3
615620

616621

622+
@unittest.skipIf(is_windows(), reason="Fails on windows")
617623
class OVTrainerImageClassificationTrainingTest(OVTrainerBaseTrainingTest):
618624
ovmodel_cls = OVModelForImageClassification
619625
task = "image-classification"
@@ -794,6 +800,7 @@ def check_ovmodel_reshaping(self, ovmodel: OVModel):
794800
}
795801

796802

803+
@unittest.skipIf(is_windows(), reason="Fails on windows")
797804
class OVTrainerAudioClassificationTrainingTest(OVTrainerBaseTrainingTest):
798805
ovmodel_cls = OVModelForAudioClassification
799806
task = "audio-classification"

0 commit comments

Comments
 (0)