From 841ecb2b8f6f4cd394692b9a34ebdbd006df341f Mon Sep 17 00:00:00 2001 From: ramenguy99 Date: Mon, 4 Sep 2023 18:02:28 +0200 Subject: [PATCH] Added AITVIEWER_INSTALL_PYQT6 environment option to setup.py --- setup.py | 53 ++++++++++++++++++++++++++++++++--------------------- 1 file changed, 32 insertions(+), 21 deletions(-) diff --git a/setup.py b/setup.py index b5e5d28..4f6bb6f 100644 --- a/setup.py +++ b/setup.py @@ -1,8 +1,39 @@ # Copyright (C) 2023 ETH Zurich, Manuel Kaufmann, Velko Vechev, Dario Mylonopoulos +import os + from setuptools import find_packages, setup from aitviewer import __version__ +INSTALL_PYQT6 = os.getenv("AITVIEWER_INSTALL_PYQT6", 0) + +requirements = [ + "torch>=1.6.0", + "numpy>=1.18,<2", + "opencv-contrib-python-headless>=4.5.1.48", + "smplx", + "moderngl-window>=2.4.3", + "moderngl>=5.8.2,<6", + "imgui==2.0.0", + "tqdm>=4.60.0", + "trimesh>=3.9.15,<4", + "scipy>=1.5.2", + "omegaconf>=2.1.1", + "roma>=1.2.3", + "joblib", + "scikit-image>=0.9.0", + "scikit-video", + "Pillow", + "websockets", + "usd-core>=23.5", +] + +# Choose PyQt version depending on environment variable. +if INSTALL_PYQT6: + requirements.append("PyQt6>=6.5.2") +else: + requirements.append("PyQt5>=5.15.4") + setup( name="aitviewer", description="Viewing and rendering of sequences of 3D data.", @@ -27,25 +58,5 @@ ], platforms=["any"], python_requires=">=3.7,<3.11", - install_requires=[ - "torch>=1.6.0", - "numpy>=1.18,<2", - "opencv-contrib-python-headless>=4.5.1.48", - "smplx", - "moderngl-window>=2.4.3", - "moderngl>=5.8.2,<6", - "PyQt5>=5.15.4", - "imgui==2.0.0", - "tqdm>=4.60.0", - "trimesh>=3.9.15,<4", - "scipy>=1.5.2", - "omegaconf>=2.1.1", - "roma>=1.2.3", - "joblib", - "scikit-image>=0.9.0", - "scikit-video", - "Pillow", - "websockets", - "usd-core>=23.5", - ], + install_requires=requirements, )