Skip to content

Commit d0bb3eb

Browse files
committed
Defer annotation evaluation via __future__ annotations import
1 parent 772906d commit d0bb3eb

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

src/llmcompressor/entrypoints/oneshot.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,8 @@
77
with various pipeline configurations for efficient model optimization.
88
"""
99

10+
from __future__ import annotations
11+
1012
import os
1113
from datetime import datetime
1214
from pathlib import Path
@@ -243,7 +245,7 @@ def oneshot(
243245
clear_sparse_session: bool = False,
244246
stage: str | None = None,
245247
# Dataset arguments
246-
dataset: "str | Dataset | DatasetDict | None" = None,
248+
dataset: str | Dataset | DatasetDict | None = None,
247249
dataset_config_name: str | None = None,
248250
dataset_path: str | None = None,
249251
splits: str | list[str] | dict[str, str] | None = None,

0 commit comments

Comments
 (0)