[project]
name = "openmind"
# version = "0.8.0.dev0"
dynamic = ["version"]
description = "openMind is a magicain who takes you to experience the mystery and creativity of AI."
readme = "README.md"
license = { text = "木兰宽松许可证， 第2版" }
authors = [
  { name = "The openMind Team" },
  { email = "dev@public.openmind.cn" },
]
dependencies = ["tqdm",
                "sentencepiece",
                "tokenizers",
                "tabulate",
                "decorator",
                "tiktoken",
                "packaging",
                "docker >= 4.34.2",
                "einops==0.8.0",
                "setuptools == 69.5.1",
                "datasets >= 2.18.0, <= 2.21.0",
                "openmind-hub == 0.9.1",
                "numpy < 2.0.0"]
requires-python = ">= 3.8, < 3.11"
classifiers = [
  "Development Status :: 1 - Planning",
  "Intended Audience :: Developers",
  "Intended Audience :: Education",
  "Intended Audience :: Science/Research",
  "Operating System :: OS Independent",
  "Programming Language :: Python :: 3",
  "Programming Language :: Python :: 3.8",
  "Programming Language :: Python :: 3.9",
  "Programming Language :: Python :: 3.10",
  "Topic :: Scientific/Engineering :: Artificial Intelligence",
  "License :: OSI Approved :: Mulan Permissive Software License v2 (MulanPSL-2.0)",
]

[project.optional-dependencies]
test = [
    "pytest",
    "pytest-cov",
    "testtools",
    "openmind[pt-cpu]",
    "pillow",
    "lm_eval == 0.4.3",
]

lint = [
    "black == 24.8.0",
    "ruff == 0.6.1",
]

ci = [
  "requests"
]

pt-cpu = [
    "torch == 2.1.0",
    "transformers >= 4.39.2, <=4.45.2",
    "accelerate >= 0.28.0, <= 1.0.0rc1",
    "evaluate == 0.4.1",
    "lm_eval == 0.4.3",
]

pt = [ 
    "openmind[pt-cpu]",
    "torch-npu == 2.1.0.post6",
]

ms = [
    "mindspore@https://ms-release.obs.cn-north-4.myhuaweicloud.com/2.4.0/MindSpore/unified/aarch64/mindspore-2.4.0-cp39-cp39-linux_aarch64.whl",
    "mindformers@git+https://gitee.com/mindspore/mindformers.git",
]

[project.urls]
Homepage = "https://gitee.com/openmind-ai/openmind"
Repository = "https://gitee.com/openmind-ai/openmind"

[build-system]
requires = ["hatchling"]
build-backend = "hatchling.build"

[tool.hatch.metadata]
allow-direct-references = true

[tool.hatch.build]
include = [
  "src/*"
]

[tool.hatch.build.targets.wheel]
packages = ["src/openmind"]

# Used to call hatch_build.py
[tool.hatch.build.targets.wheel.hooks.custom]

# Used to get version information
[tool.hatch.metadata.hooks.custom]
#path = "custom_metadata_hook.py"

[tool.black]
line-length = 120
target-version = ['py38']
include = '\.pyi?$'

[tool.ruff]
line-length = 120
# Enables the isort rules on top of the default rules

# "F401": Ignore import vilations in all `__init__.py` files.
# "I": Ignore isort in all `__init__.py` files to avoid circle import.
[tool.ruff.lint.per-file-ignores]
"__init__.py" = ["F401", "I"]

[tool.ruff.lint.isort]
lines-after-imports = 2
# Sort the imports by module, independent for import style (from / import).
force-sort-within-sections = true

[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"

[project.scripts]
openmind-cli = "openmind._cli.openmind_cli:main"