pyproject.toml

[build-system]
requires = ["cython>=3.0", "pip>=23.0", "setuptools>=61.0", "wheel"]

[project]
name = "mg2hfbot"
requires-python = ">=3.10,<3.11"
version = "0.1.0"
description = "Use lerobot to convert the mimicgen data and train/test on robosuite envs"
classifiers = [
	"Natural Language :: English",
	"Operating System :: POSIX :: Linux",
	"Operating System :: MacOS :: MacOS X",
	"Programming Language :: Python",
	"Programming Language :: Python :: 3.10",
]

dependencies = [
    "gymnasium==0.29.1",
    "mimicgen@git+https://github.com/kywch/mimicgen.git",
    "mujoco==2.3.2",  # see https://mimicgen.github.io/docs/introduction/installation.html#mujoco
    "robosuite==1.4.1",
    "robomimic@git+https://github.com/ARISE-Initiative/robomimic.git",
    "robosuite_task_zoo@git+https://github.com/kywch/robosuite-task-zoo.git",
    "lerobot@git+https://github.com/kywch/lerobot.git",
    "wandb",
    "torch<2.5",
]

[tool.pixi.project]
channels = ["pytorch", "nvidia", "conda-forge"]
platforms = ["linux-64"]

[system-requirements]
cuda = "12.1"

[tool.pixi.dependencies]
cuda-toolkit = "12.1"
cmake = ">=3.30.5,<4"
evdev = ">=1.7.1,<2"
make = ">=4.4.1,<5"
numba = ">=0.59.0"
pyarrow = ">=17.0.0"
pytorch = ">=2.0,<2.5"
pytorch-cuda = "12.1"
rerun-sdk = ">=0.15.1"

# For convenience
ruff = "*"
pre-commit = "*"

[tool.pixi.pypi-dependencies]
mg2hfbot = { path = ".", editable = true }

[tool.pixi.tasks]
train = "python train.py"
eval = "python eval.py"
convert = "python convert_to_lerobot.py"

[tool.setuptools.packages.find]
where = ["."]
exclude = ["tests"]

[tool.distutils.bdist_wheel]
universal = true

[tool.ruff]
# Exclude a variety of commonly ignored directories.
exclude = [
    ".bzr",
    ".direnv",
    ".eggs",
    ".git",
    ".git-rewrite",
    ".hg",
    ".ipynb_checkpoints",
    ".mypy_cache",
    ".nox",
    ".pants.d",
    ".pyenv",
    ".pytest_cache",
    ".pytype",
    ".ruff_cache",
    ".svn",
    ".tox",
    ".venv",
    ".vscode",
    "__pypackages__",
    "_build",
    "buck-out",
    "build",
    "dist",
    "node_modules",
    "site-packages",
    "venv",
]

# Same as Black.
line-length = 100
indent-width = 4

# Assume Python 3.10
target-version = "py310"

[tool.ruff.lint]
# Enable Pyflakes (`F`) and a subset of the pycodestyle (`E`)  codes by default.
# Unlike Flake8, Ruff doesn't enable pycodestyle warnings (`W`) or
# McCabe complexity (`C901`) by default.
select = ["E4", "E7", "E9", "F"]
ignore = []

# Allow fix for all enabled rules (when `--fix`) is provided.
fixable = ["ALL"]
unfixable = []

# Allow unused variables when underscore-prefixed.
dummy-variable-rgx = "^(_+|(_+[a-zA-Z0-9_]*[a-zA-Z0-9]+?))$"

[tool.ruff.format]
# Like Black, use double quotes for strings.
quote-style = "double"

# Like Black, indent with spaces, rather than tabs.
indent-style = "space"

# Like Black, respect magic trailing commas.
skip-magic-trailing-comma = false

# Like Black, automatically detect the appropriate line ending.
line-ending = "auto"

# Enable auto-formatting of code examples in docstrings. Markdown,
# reStructuredText code/literal blocks and doctests are all supported.
#
# This is currently disabled by default, but it is planned for this
# to be opt-out in the future.
docstring-code-format = false

# Set the line length limit used when formatting code snippets in
# docstrings.
#
# This only has an effect when the `docstring-code-format` setting is
# enabled.
docstring-code-line-length = "dynamic"