diff --git a/pyproject.toml b/pyproject.toml index 5c4a9ad..1f91b14 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -95,6 +95,9 @@ mlflow = ["mlflow>=3.1.1"] gradio = [ "gradio>=6.9.0", ] +dvc = [ + "dvclive>=3.49.0", +] [tool.ruff] line-length = 79 diff --git a/tests/test_models/test_backbones.py b/tests/test_models/test_backbones.py index 8565a07..6f26dad 100644 --- a/tests/test_models/test_backbones.py +++ b/tests/test_models/test_backbones.py @@ -96,7 +96,7 @@ def test_registry_build_unknown_name_raises(): name = "NonExistentBackbone" with pytest.raises(NotImplementedError): - backbone_registry.build(FakeConfig()) # type: ignore[arg-type] + backbone_registry.build(FakeConfig()) # ty: ignore[invalid-argument-type] def test_backbone_config_validates_unet_from_dict(): diff --git a/tests/test_targets/test_targets.py b/tests/test_targets/test_targets.py index aa8433f..72c9c7a 100644 --- a/tests/test_targets/test_targets.py +++ b/tests/test_targets/test_targets.py @@ -1,7 +1,6 @@ from collections.abc import Callable from pathlib import Path -import pytest from soundevent import data, terms from batdetect2.targets import TargetConfig, build_roi_mapping, build_targets