Skip to content

Commit a23a90b

Browse files
dependabot[bot]shchekleinDave Berenbaum
authored
bump mypy, fixing tests and deps (#832)
* build(deps): bump mypy from 1.10.0 to 1.10.1 Bumps [mypy](https://github.com/python/mypy) from 1.10.0 to 1.10.1. - [Changelog](https://github.com/python/mypy/blob/master/CHANGELOG.md) - [Commits](python/mypy@v1.10.0...v1.10.1) --- updated-dependencies: - dependency-name: mypy dependency-type: direct:production update-type: version-update:semver-patch ... Signed-off-by: dependabot[bot] <[email protected]> * remove aliase classes * drop catalyst * make hf tests work (#833) --------- Signed-off-by: dependabot[bot] <[email protected]> Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com> Co-authored-by: Ivan Shcheklein <[email protected]> Co-authored-by: Dave Berenbaum <[email protected]>
1 parent 39ad649 commit a23a90b

File tree

5 files changed

+8
-122
lines changed

5 files changed

+8
-122
lines changed

pyproject.toml

+5-4
Original file line numberDiff line numberDiff line change
@@ -55,24 +55,25 @@ tests = [
5555
"dvclive[image,plots,markdown]",
5656
"ipython",
5757
"pytest_voluptuous",
58-
"dpath"
58+
"dpath",
59+
"transformers[torch]",
60+
"tf-keras"
5961
]
6062
dev = [
6163
"dvclive[all,tests]",
62-
"mypy==1.10.0",
64+
"mypy==1.10.1",
6365
"types-PyYAML"
6466
]
6567
mmcv = ["mmcv"]
6668
tf = ["tensorflow"]
6769
xgb = ["xgboost"]
6870
lgbm = ["lightgbm"]
6971
huggingface = ["transformers", "datasets"]
70-
catalyst = ["catalyst>22"]
7172
fastai = ["fastai"]
7273
lightning = ["lightning>=2.0", "torch", "jsonargparse[signatures]>=4.26.1"]
7374
optuna = ["optuna"]
7475
all = [
75-
"dvclive[image,mmcv,tf,xgb,lgbm,huggingface,catalyst,fastai,lightning,optuna,plots,markdown]"
76+
"dvclive[image,mmcv,tf,xgb,lgbm,huggingface,fastai,lightning,optuna,plots,markdown]"
7677
]
7778

7879
[project.urls]

src/dvclive/catalyst.py

-23
This file was deleted.

src/dvclive/plots/utils.py

+1-2
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,6 @@
11
import json
22

33
NUMPY_INTS = [
4-
"int_",
54
"intc",
65
"intp",
76
"int8",
@@ -13,7 +12,7 @@
1312
"uint32",
1413
"uint64",
1514
]
16-
NUMPY_FLOATS = ["float_", "float16", "float32", "float64"]
15+
NUMPY_FLOATS = ["float16", "float32", "float64"]
1716
NUMPY_SCALARS = NUMPY_INTS + NUMPY_FLOATS
1817

1918

tests/frameworks/test_catalyst.py

-92
This file was deleted.

tests/frameworks/test_huggingface.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -103,6 +103,7 @@ def args():
103103
num_train_epochs=2,
104104
save_strategy="epoch",
105105
report_to="none", # Disable auto-reporting to avoid duplication
106+
use_cpu=True,
106107
)
107108

108109

@@ -173,7 +174,7 @@ def test_huggingface_log_model(
173174
trainer.train()
174175

175176
expected_call_count = {
176-
"all": 2,
177+
"all": 3,
177178
True: 1,
178179
False: 0,
179180
None: 0,

0 commit comments

Comments
 (0)