Skip to content

Commit 8ffa626

Browse files
committed
Merge branch 'main' into vocquant
2 parents c518362 + 3c09887 commit 8ffa626

5 files changed

Lines changed: 7 additions & 5 deletions

File tree

.github/workflows/ci.yaml

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -10,14 +10,16 @@ jobs:
1010
strategy:
1111
matrix:
1212
os: ["ubuntu-latest", "windows-latest"]
13-
python-version: ["3.9", "3.10", "3.11", "3.12"]
13+
python-version: ["3.9", "3.10", "3.11", "3.12", "3.13"]
1414
exclude:
1515
- os: windows-latest
1616
python-version: "3.9"
1717
- os: windows-latest
1818
python-version: "3.11"
1919
- os: windows-latest
2020
python-version: "3.12"
21+
- os: windows-latest
22+
python-version: "3.13"
2123
fail-fast: false
2224

2325
steps:
@@ -55,7 +57,7 @@ jobs:
5557
# Run tests with coverage
5658
- name: Run tests under coverage
5759
run: |
58-
coverage run -m pytest
60+
coverage run --source=model2vec -m pytest
5961
coverage report
6062
6163
# Upload results to Codecov

Makefile

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,6 @@ install:
1010

1111
install-no-pre-commit:
1212
uv pip install ".[dev,distill,inference,train]"
13-
uv pip install "torch<2.5.0"
1413

1514
install-base:
1615
uv sync --extra dev

README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -92,7 +92,7 @@ m2v_model.save_pretrained("m2v_model")
9292
After distillation, you can also fine-tune your own classification models on top of the distilled model, or on a pre-trained model. First, make sure you install the `training` extras with:
9393

9494
```bash
95-
pip install model2vec[training]
95+
pip install model2vec[train]
9696
```
9797

9898
Then, you can fine-tune a model as follows:

model2vec/train/README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@ We support both single and multi-label classification, which work seamlessly bas
99
To train, make sure you install the training extra:
1010

1111
```
12-
pip install model2vec[training]
12+
pip install model2vec[train]
1313
```
1414

1515
# Quickstart

pyproject.toml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,7 @@ classifiers = [
1919
"Programming Language :: Python :: 3.10",
2020
"Programming Language :: Python :: 3.11",
2121
"Programming Language :: Python :: 3.12",
22+
"Programming Language :: Python :: 3.13",
2223
"Natural Language :: English",
2324
]
2425

0 commit comments

Comments
 (0)