File size: 2,771 Bytes
0037d53 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 | # Copyright (c) 2024-2026, The UW Lab Project Developers. (https://github.com/uw-lab/UWLab/blob/main/CONTRIBUTORS.md).
# All Rights Reserved.
#
# SPDX-License-Identifier: BSD-3-Clause
"""Installation script for the 'uwlab_tasks' python package."""
import os
import platform
import sys
import toml
from setuptools import setup
# Obtain the extension data from the extension.toml file
EXTENSION_PATH = os.path.dirname(os.path.realpath(__file__))
# Read the extension.toml file
EXTENSION_TOML_DATA = toml.load(os.path.join(EXTENSION_PATH, "config", "extension.toml"))
# Minimum dependencies required prior to installation
INSTALL_REQUIRES = [
# OmniReset dependencies
"rtree",
"zarr==2.18.3",
"numcodecs==0.13.1",
"cmaes",
]
is_linux_x86_64 = platform.system() == "Linux" and platform.machine() in ("x86_64", "AMD64")
py = f"cp{sys.version_info.major}{sys.version_info.minor}"
wheel_by_py = {
"cp311": (
"https://github.com/MiroPsota/torch_packages_builder/releases/download/pytorch3d-0.7.8/"
"pytorch3d-0.7.8%2Bpt2.7.0cu128-cp311-cp311-linux_x86_64.whl"
),
"cp310": (
"https://github.com/MiroPsota/torch_packages_builder/releases/download/pytorch3d-0.7.8/"
"pytorch3d-0.7.8%2Bpt2.7.0cu128-cp310-cp310-linux_x86_64.whl"
),
}
if is_linux_x86_64 and py in wheel_by_py:
INSTALL_REQUIRES.append(f"pytorch3d @ {wheel_by_py[py]}")
is_linux_x86_64 = platform.system() == "Linux" and platform.machine() in ("x86_64", "AMD64")
py = f"cp{sys.version_info.major}{sys.version_info.minor}"
wheel_by_py = {
"cp311": (
"https://github.com/MiroPsota/torch_packages_builder/releases/download/pytorch3d-0.7.8/"
"pytorch3d-0.7.8%2Bpt2.7.0cu128-cp311-cp311-linux_x86_64.whl"
),
"cp310": (
"https://github.com/MiroPsota/torch_packages_builder/releases/download/pytorch3d-0.7.8/"
"pytorch3d-0.7.8%2Bpt2.7.0cu128-cp310-cp310-linux_x86_64.whl"
),
}
if is_linux_x86_64 and py in wheel_by_py:
INSTALL_REQUIRES.append(f"pytorch3d @ {wheel_by_py[py]}")
# Installation operation
setup(
name="uwlab_tasks",
author="UW and Isaac Lab Project Developers",
maintainer="UW and Isaac Lab Project Developers",
url=EXTENSION_TOML_DATA["package"]["repository"],
version=EXTENSION_TOML_DATA["package"]["version"],
description=EXTENSION_TOML_DATA["package"]["description"],
keywords=EXTENSION_TOML_DATA["package"]["keywords"],
license="BSD-3-Clause",
include_package_data=True,
python_requires=">=3.10",
install_requires=INSTALL_REQUIRES,
packages=["uwlab_tasks"],
classifiers=[
"Natural Language :: English",
"Programming Language :: Python :: 3.10",
"Isaac Sim :: 4.5.0",
],
zip_safe=False,
)
|