Skip to content

Commit 82c27dd

Browse files
committed
Fixed pre-ci test error
Signed-off-by: Cheng, Penghui <penghui.cheng@intel.com>
1 parent e804df3 commit 82c27dd

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

optimum/intel/neural_compressor/quantization.py

+3-3
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@
1919
from enum import Enum
2020
from itertools import chain
2121
from pathlib import Path
22-
from typing import Callable, Dict, Optional, TypeAlias, Union
22+
from typing import Callable, Dict, Optional, Union
2323

2424
import torch
2525
from datasets import Dataset, load_dataset
@@ -81,9 +81,9 @@
8181
from intel_extension_for_transformers.llm.quantization.utils import convert_to_quantized_model
8282
from intel_extension_for_transformers.transformers.utils.config import WeightOnlyQuantConfig
8383

84-
Config: TypeAlias = Union[PostTrainingQuantConfig, WeightOnlyQuantConfig]
84+
Config = Union[PostTrainingQuantConfig, WeightOnlyQuantConfig]
8585
else:
86-
Config: TypeAlias = PostTrainingQuantConfig
86+
Config = PostTrainingQuantConfig
8787

8888
logger = logging.getLogger(__name__)
8989

0 commit comments

Comments
 (0)