diff --git a/setup.py b/setup.py index bb44c5c..ed83058 100644 --- a/setup.py +++ b/setup.py @@ -19,7 +19,7 @@ DOWNLOAD_URL = "https://github.com/si-cim/prototorch_models.git" with open("README.md", "r") as fh: long_description = fh.read() -INSTALL_REQUIRES = ["prototorch>=0.4.1", "pytorch_lightning", "torchmetrics"] +INSTALL_REQUIRES = ["prototorch>=0.4.2", "pytorch_lightning", "torchmetrics"] DEV = ["bumpversion"] EXAMPLES = ["matplotlib", "scikit-learn"] TESTS = ["codecov", "pytest"]