|
18 | 18 | import setuptools |
19 | 19 | from setuptools_scm import get_version |
20 | 20 |
|
21 | | -# Package configuration ############################################################################ |
22 | | -name = "nvidia-modelopt" |
23 | | -# TODO: Set version to static stable release version when creating the release branch |
24 | | -# version = os.environ.get("SETUPTOOLS_SCM_PRETEND_VERSION", "X.Y.Z") |
| 21 | +# TODO: Set fallback_version to X.Y.Z release version when creating the release branch |
25 | 22 | version = get_version(root=".", fallback_version="0.0.0") |
26 | | -packages = setuptools.find_namespace_packages(include=["modelopt*"]) |
27 | | -package_dir = {"": "."} |
28 | | -package_data = {"modelopt": ["**/*.h", "**/*.cpp", "**/*.cu"]} |
29 | | -setup_kwargs = {} |
30 | 23 |
|
31 | 24 | # Required and optional dependencies ############################################################### |
32 | 25 | required_deps = [ |
|
103 | 96 | # build/packaging tools |
104 | 97 | "dev-build": [ |
105 | 98 | "cython", |
106 | | - "setuptools>=67.8.0", |
107 | | - "setuptools_scm>=7.1.0", |
108 | | - "twine", |
| 99 | + "setuptools>=80", |
| 100 | + "setuptools-scm>=8", |
109 | 101 | ], |
110 | 102 | } |
111 | 103 |
|
|
118 | 110 |
|
119 | 111 | if __name__ == "__main__": |
120 | 112 | setuptools.setup( |
121 | | - name=name, |
| 113 | + name="nvidia-modelopt", |
122 | 114 | version=version, |
123 | 115 | description="Nvidia TensorRT Model Optimizer: a unified model optimization and deployment toolkit.", |
124 | 116 | long_description="Checkout https://github.com/nvidia/TensorRT-Model-Optimizer for more information.", |
|
136 | 128 | python_requires=">=3.10,<3.13", |
137 | 129 | install_requires=required_deps, |
138 | 130 | extras_require=optional_deps, |
139 | | - packages=packages, |
140 | | - package_dir=package_dir, |
141 | | - package_data=package_data, |
142 | | - **setup_kwargs, |
| 131 | + packages=setuptools.find_namespace_packages(include=["modelopt*"]), |
| 132 | + package_dir={"": "."}, |
| 133 | + package_data={"modelopt": ["**/*.h", "**/*.cpp", "**/*.cu"]}, |
143 | 134 | ) |
0 commit comments