diff --git a/pesto/__init__.py b/pesto/__init__.py index cff2aebc8285d3f463252869e8f432cb0189bade..5da69bbe1733e6b3bb286735d0dd2a98613ce66a 100644 --- a/pesto/__init__.py +++ b/pesto/__init__.py @@ -1 +1,4 @@ from .core import load_model, predict, predict_from_files + + +__version__ = '1.1.0' diff --git a/pesto/core.py b/pesto/core.py index 7c3522510d142d98d7df2310e6f337d468f6dd6f..dd28a5872e8caa8a1f49160743b9d9fe05138a42 100644 --- a/pesto/core.py +++ b/pesto/core.py @@ -146,4 +146,3 @@ def predict_from_files( predictions = [p.cpu().numpy() for p in predictions] for fmt in export_format: export(fmt, output_file, *predictions) - diff --git a/pyproject.toml b/pyproject.toml index 6464d322af62df07aeccc2ccf85ecdbde83a636b..b2fb7d05dcc35b499ede3b0562769602baaab66d 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta" [project] name = "pesto-pitch" -version = "1.0.0" +dynamic = ["version"] authors = [ {name = "Alain Riou", email = "alain.riou@sony.com"} ] @@ -41,5 +41,8 @@ source = "https://github.com/SonyCSLParis/pesto" [tool.pytest.ini_options] testpaths = "tests/" +[tool.setuptools.dynamic] +version = {attr = "pesto.__version__"} + [tool.setuptools.package-data] pesto = ["weights/*.ckpt"]