From 48f4e8f673128935af64c82c557a8131190a0d4b Mon Sep 17 00:00:00 2001 From: Jack Wolfard <31635014+JackWolfard@users.noreply.github.com> Date: Wed, 2 Aug 2023 22:25:11 -0700 Subject: [PATCH] Add packaging as an install dependency (#2369) Needed by `torch_mlir._version`. Resolves #2368. --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index 68d544948..6b8484e05 100644 --- a/setup.py +++ b/setup.py @@ -163,7 +163,7 @@ setup( ext_modules=[ CMakeExtension("torch_mlir._mlir_libs._jit_ir_importer"), ] if not TORCH_MLIR_ENABLE_ONLY_MLIR_PYTHON_BINDINGS else [CMakeExtension("torch_mlir._mlir_libs._torchMlir")], - install_requires=["numpy", ] + ( + install_requires=["numpy", "packaging"] + ( [f"torch=={torch.__version__}".split("+", 1)[0], ] if not TORCH_MLIR_ENABLE_ONLY_MLIR_PYTHON_BINDINGS else []), zip_safe=False, )