fix nightly ut (intel#1125)

mengfei25 · Jan 10, 2024 · fdefb9d · fdefb9d
1 parent 938a3f0
commit fdefb9d
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 1 deletion.
diff --git a/tests/Nightly/test_finetuning_data.py → ...ightly/finetuning/test_finetuning_data.py b/tests/Nightly/test_finetuning_data.py → ...ightly/finetuning/test_finetuning_data.py
@@ -1,3 +1,20 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+#
+# Copyright (c) 2023 Intel Corporation
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
 import os
 import unittest
 

diff --git a/tests/CI/test_quantization_qa_ipex.py b/tests/CI/test_quantization_qa_ipex.py
@@ -8,8 +8,11 @@
 os.environ["CUDA_VISIBLE_DEVICES"] = ""
 
 # example test for question-answering quantization with IPEX only for now
+EXAMPLE_PATH="../../examples/huggingface/pytorch/"
+if not os.path.exists(EXAMPLE_PATH):
+    EXAMPLE_PATH="../examples/huggingface/pytorch/"
 SRC_DIRS = [
-    os.path.join("../../examples/huggingface/pytorch/", dirname)
+    os.path.join(EXAMPLE_PATH, dirname)
     for dirname in [
         "question-answering/quantization/",
     ]