|
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