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]>
iterative · Jul 10, 2024 · a23a90b · a23a90b
1 parent 39ad649
commit a23a90b
Show file tree

Hide file tree

Showing 5 changed files with 8 additions and 122 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -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]

diff --git a/src/dvclive/catalyst.py b/src/dvclive/catalyst.py
diff --git a/src/dvclive/plots/utils.py b/src/dvclive/plots/utils.py
@@ -1,7 +1,6 @@
 import json
 
 NUMPY_INTS = [
-    "int_",
     "intc",
     "intp",
     "int8",
@@ -13,7 +12,7 @@
     "uint32",
     "uint64",
 ]
-NUMPY_FLOATS = ["float_", "float16", "float32", "float64"]
+NUMPY_FLOATS = ["float16", "float32", "float64"]
 NUMPY_SCALARS = NUMPY_INTS + NUMPY_FLOATS
 
 

diff --git a/tests/frameworks/test_catalyst.py b/tests/frameworks/test_catalyst.py
diff --git a/tests/frameworks/test_huggingface.py b/tests/frameworks/test_huggingface.py
@@ -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,
     )
 
 
@@ -173,7 +174,7 @@ def test_huggingface_log_model(
     trainer.train()
 
     expected_call_count = {
-        "all": 2,
+        "all": 3,
         True: 1,
         False: 0,
         None: 0,