Skip to content

Commit a4ee2fc

Browse files
committed
Fix docstring and type hints for recipe and splits parameters respectively
1 parent cae4dd8 commit a4ee2fc

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

src/llmcompressor/entrypoints/oneshot.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -241,7 +241,7 @@ def oneshot(
241241
dataset: str | "Dataset" | "DatasetDict" | None = None,
242242
dataset_config_name: str | None = None,
243243
dataset_path: str | None = None,
244-
splits: str | list | dict | None = None,
244+
splits: str | list[str] | dict[str, str] | None = None,
245245
num_calibration_samples: int = 512,
246246
shuffle_calibration_samples: bool = True,
247247
max_seq_length: int = 384,
@@ -287,7 +287,8 @@ def oneshot(
287287
tag, or commit id).
288288
289289
# Recipe arguments
290-
:param recipe: Path to a LLM Compressor sparsification recipe.
290+
:param recipe: Path to a LLM Compressor sparsification recipe, or a list of paths
291+
to multiple recipes.
291292
:param recipe_args: List of recipe arguments to evaluate, in the
292293
format "key1=value1", "key2=value2".
293294
:param clear_sparse_session: Whether to clear CompressionSession/

0 commit comments

Comments
 (0)