drop catalyst

iterative · Jul 1, 2024 · 9ea9f75 · 9ea9f75
1 parent b9e1c46
commit 9ea9f75
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 119 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -55,7 +55,9 @@ tests = [
   "dvclive[image,plots,markdown]",
   "ipython",
   "pytest_voluptuous",
-  "dpath"
+  "dpath",
+  "transformers[torch]",
+  "keras<3"
 ]
 dev = [
   "dvclive[all,tests]",
@@ -67,12 +69,11 @@ 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/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
@@ -173,7 +173,7 @@ def test_huggingface_log_model(
     trainer.train()
 
     expected_call_count = {
-        "all": 2,
+        "all": 3,
         True: 1,
         False: 0,
         None: 0,