File size: 2,934 Bytes
53c10a4 | 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 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 | [build-system]
requires = ["setuptools>=77", "wheel"]
build-backend = "setuptools.build_meta"
[project]
name = "orarl"
version = "0.1.0"
description = "Annotations as rollouts for unified video MLLM reinforcement learning"
readme = "README.md"
requires-python = ">=3.10"
license = "Apache-2.0"
license-files = ["LICENSE", "NOTICE"]
keywords = [
"reinforcement-learning",
"video-understanding",
"multimodal",
"grpo",
]
authors = [
{ name = "Yunheng Li" },
{ name = "Guohong Mu" },
{ name = "Hao Li" },
{ name = "Shengsheng Qian" },
{ name = "Dingwen Zhang" },
{ name = "Qibin Hou" },
{ name = "Ming-Ming Cheng" },
]
classifiers = [
"Development Status :: 3 - Alpha",
"Programming Language :: Python :: 3",
"Programming Language :: Python :: 3.10",
"Programming Language :: Python :: 3.11",
"Programming Language :: Python :: 3.12",
]
dependencies = [
"numpy",
"PyYAML",
"torch",
]
[project.optional-dependencies]
test = ["pytest"]
lint = ["ruff"]
hf = ["huggingface_hub"]
[project.urls]
Homepage = "https://orarl.github.io/"
Documentation = "https://orarl.github.io/"
[project.scripts]
orarl-train = "orarl.cli.train:main"
orarl-eval = "orarl.cli.evaluate:main"
orarl-eval-data = "orarl.cli.eval_data:main"
orarl-prepare = "orarl.cli.prepare:main"
[tool.setuptools.packages.find]
where = ["."]
include = ["orarl*", "verl*"]
[tool.setuptools.data-files]
"share/orarl" = ["environment.yml", "requirements-cu129.txt"]
"share/orarl/configs" = ["configs/*.yaml"]
"share/orarl/docs" = ["docs/*.md"]
"share/orarl/scripts" = ["scripts/*.py", "scripts/*.sh"]
"share/orarl/eval" = ["eval/README.md"]
"share/orarl/eval/task" = ["eval/task/*.py", "eval/task/*.sh"]
"share/orarl/eval/task/mindcube" = ["eval/task/mindcube/*.py"]
"share/orarl/eval/task/mmsi" = ["eval/task/mmsi/*.py", "eval/task/mmsi/*.sh"]
"share/orarl/eval/task/revsi" = ["eval/task/revsi/*.py", "eval/task/revsi/*.sh"]
"share/orarl/eval/task/segmentation" = [
"eval/task/segmentation/*.py",
"eval/task/segmentation/*.sh",
]
"share/orarl/eval/task/spatial_grounding" = ["eval/task/spatial_grounding/*.py"]
"share/orarl/eval/task/spatial_temporal_grounding" = [
"eval/task/spatial_temporal_grounding/*.py",
]
"share/orarl/eval/task/temporal_grounding" = [
"eval/task/temporal_grounding/*.py",
"eval/task/temporal_grounding/*.sh",
]
"share/orarl/eval/task/tracking" = ["eval/task/tracking/*.py"]
[tool.pytest.ini_options]
testpaths = ["tests"]
addopts = "-ra"
[tool.ruff]
target-version = "py310"
line-length = 100
# The evaluation runtime, the training runtime, and the checkpoint merger are
# ported verbatim from the stack that produced the published numbers. Several of
# their long lines are prompt literals, so reformatting them would change model
# inputs.
extend-exclude = ["eval/task", "verl", "scripts/model_merger.py"]
[tool.ruff.lint]
select = ["E", "F", "I", "W"]
|