Skip to content

Commit

Permalink
Skip training tests on windows (#930)
Browse files Browse the repository at this point in the history
* Skip training tests on windows

* Trigger Tests

* Trigger Tests
  • Loading branch information
nikita-savelyevv authored Oct 23, 2024
1 parent 966c8c2 commit 227defe
Showing 1 changed file with 7 additions and 0 deletions.
7 changes: 7 additions & 0 deletions tests/openvino/test_training.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,6 +12,7 @@
# See the License for the specific language governing permissions and
# limitations under the License.

import os
import random
import re
import shutil
Expand Down Expand Up @@ -89,6 +90,10 @@ def initialize_movement_sparsifier_parameters_by_sparsity(
operand.bias_importance.copy_(bias_init_tensor)


def is_windows():
return os.name == "nt"


def is_avx_vnni_supported() -> bool:
return any(re.search("avx.*vnni", flag.lower()) is not None for flag in cpuinfo.get_cpu_info()["flags"])

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


@unittest.skipIf(is_windows(), reason="Fails on windows")
class OVTrainerImageClassificationTrainingTest(OVTrainerBaseTrainingTest):
ovmodel_cls = OVModelForImageClassification
task = "image-classification"
Expand Down Expand Up @@ -794,6 +800,7 @@ def check_ovmodel_reshaping(self, ovmodel: OVModel):
}


@unittest.skipIf(is_windows(), reason="Fails on windows")
class OVTrainerAudioClassificationTrainingTest(OVTrainerBaseTrainingTest):
ovmodel_cls = OVModelForAudioClassification
task = "audio-classification"
Expand Down

0 comments on commit 227defe

Please sign in to comment.