Skip to content

Commit

Permalink
bump mypy, fixing tests and deps (#832)
Browse files Browse the repository at this point in the history
* 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] <support@github.com>

* remove aliase classes

* drop catalyst

* make hf tests work (#833)

---------

Signed-off-by: dependabot[bot] <support@github.com>
Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>
Co-authored-by: Ivan Shcheklein <shcheklein@gmail.com>
Co-authored-by: Dave Berenbaum <dave@iterative.ai>
  • Loading branch information
3 people committed Jul 10, 2024
1 parent 39ad649 commit a23a90b
Show file tree
Hide file tree
Showing 5 changed files with 8 additions and 122 deletions.
9 changes: 5 additions & 4 deletions pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -55,24 +55,25 @@ tests = [
"dvclive[image,plots,markdown]",
"ipython",
"pytest_voluptuous",
"dpath"
"dpath",
"transformers[torch]",
"tf-keras"
]
dev = [
"dvclive[all,tests]",
"mypy==1.10.0",
"mypy==1.10.1",
"types-PyYAML"
]
mmcv = ["mmcv"]
tf = ["tensorflow"]
xgb = ["xgboost"]
lgbm = ["lightgbm"]
huggingface = ["transformers", "datasets"]
catalyst = ["catalyst>22"]
fastai = ["fastai"]
lightning = ["lightning>=2.0", "torch", "jsonargparse[signatures]>=4.26.1"]
optuna = ["optuna"]
all = [
"dvclive[image,mmcv,tf,xgb,lgbm,huggingface,catalyst,fastai,lightning,optuna,plots,markdown]"
"dvclive[image,mmcv,tf,xgb,lgbm,huggingface,fastai,lightning,optuna,plots,markdown]"
]

[project.urls]
Expand Down
23 changes: 0 additions & 23 deletions src/dvclive/catalyst.py

This file was deleted.

3 changes: 1 addition & 2 deletions src/dvclive/plots/utils.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,6 @@
import json

NUMPY_INTS = [
"int_",
"intc",
"intp",
"int8",
Expand All @@ -13,7 +12,7 @@
"uint32",
"uint64",
]
NUMPY_FLOATS = ["float_", "float16", "float32", "float64"]
NUMPY_FLOATS = ["float16", "float32", "float64"]
NUMPY_SCALARS = NUMPY_INTS + NUMPY_FLOATS


Expand Down
92 changes: 0 additions & 92 deletions tests/frameworks/test_catalyst.py

This file was deleted.

3 changes: 2 additions & 1 deletion tests/frameworks/test_huggingface.py
Original file line number Diff line number Diff line change
Expand Up @@ -103,6 +103,7 @@ def args():
num_train_epochs=2,
save_strategy="epoch",
report_to="none", # Disable auto-reporting to avoid duplication
use_cpu=True,
)


Expand Down Expand Up @@ -173,7 +174,7 @@ def test_huggingface_log_model(
trainer.train()

expected_call_count = {
"all": 2,
"all": 3,
True: 1,
False: 0,
None: 0,
Expand Down

0 comments on commit a23a90b

Please sign in to comment.