Skip to content

Commit c99b98c

Browse files
committed
Depcrecated apply_zipf and use_subword parameters
1 parent 9056318 commit c99b98c

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/test_distillation.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -48,7 +48,7 @@ def test_distill_from_model(
4848
pca_dims: int | None,
4949
sif_coefficient: float | None,
5050
) -> None:
51-
"""Test distill_from_model vs distill produce consistent outputs."""
51+
"""Test distill function with different parameters."""
5252
# Mock the return value of model_info to avoid calling the Hugging Face API
5353
mock_model_info.return_value = type("ModelInfo", (object,), {"cardData": {"language": "en"}})
5454
mock_auto_model.return_value = mock_transformer
@@ -112,7 +112,7 @@ def test_distill_removal_pattern(
112112
)
113113
assert len(static_model.embedding) == expected_vocab_size
114114

115-
# Weird/bad regex should raise
115+
# Weird pattern.
116116
with pytest.raises(ValueError):
117117
_ = distill_from_model(
118118
model=mock_transformer,

0 commit comments

Comments
 (0)