Skip to content

Make tensorflow an optional dependency (vus[models] extra)#7

Open
martinuray wants to merge 2 commits into
thedatumorg:mainfrom
martinuray:main
Open

Make tensorflow an optional dependency (vus[models] extra)#7
martinuray wants to merge 2 commits into
thedatumorg:mainfrom
martinuray:main

Conversation

@martinuray

Copy link
Copy Markdown

This PR is addressing issue #6 .

TensorFlow was a mandatory dependency for the whole package, even though it's only needed by three of the baseline detectors under vus.models: lstm.py, AE.py, and cnn.py. The metrics themselves, which are the actual contribution of the paper, have no TensorFlow or GPU dependency at all.

This meant every pip install vus pulled in TensorFlow, which is a common source of dependency conflicts (TensorFlow/PyTorch/CUDA version mismatches) in projects that don't otherwise use it, even when the only thing needed from vus is the evaluation metrics.

This PR moves tensorflow out of install_requires and into an optional models extra. pip install vus now installs a lightweight package with just the metrics, and pip install vus[models] additionally installs TensorFlow for the three baseline detectors that need it.

This is technically a breaking change for anyone relying on a plain pip install vus to also pull in TensorFlow implicitly, but should be a net improvement for most users, who only need the metrics and now get a lighter install with one less source of dependency conflicts.

Martin Uray added 2 commits July 2, 2026 20:35
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant