PoET-2/pyproject.toml at main · OpenProteinAI/PoET-2 · GitHub

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
[project]
authors = [
  { name = "Tristan Bepler", email = "tbepler@gmail.com" },
  { name = "Timothy Fei Truong Jr", email = "ttruong@ne47.bio" },
]
dependencies = [
  "torch==2.6",
  "flash-attn @ https://github.com/OpenProteinAI/flash-attention/releases/download/v2.6.3-alibi-as-bh-bias-16/flash_attn-2.6.3+cu118torch2.6cxx11abiFALSE-cp310-cp310-linux_x86_64.whl",
  "numpy>=2.2.5,<3",
  "tqdm>=4.67.1,<5",
  "pydantic>=2.11.4,<3",
  "numba>=0.61.2,<0.62",
  "xformers==0.0.29.post3",
  "setuptools>=80.9.0,<81",
]
name = "PoET-2"
requires-python = ">=3.10,<3.11"
version = "0.1.0"

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

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

[project.urls]
"Homepage" = "https://github.com/OpenProteinAI/PoET-2"
"Bug Tracker" = "https://github.com/OpenProteinAI/PoET-2/issues"

[tool.pixi.workspace]
channels = ["conda-forge"]
platforms = ["linux-64"]

[tool.pixi.pypi-dependencies]
requests = ">=2.32.4,<3"
scipy = ">=1.15.2,<2"
pandas = ">=2.2.3,<3"
scikit-learn = ">=1.6.1,<2"
matplotlib = ">=3.10.1,<4"
seaborn = ">=0.13.2,<0.14"
pyarrow = ">=20.0.0,<21"
ipykernel = ">=6.29.5,<7"
pyzstd = ">=0.16.2,<0.17"
openprotein-python = "==0.7.0"
poet_2 = { path = ".", editable = true }

[tool.pixi.pypi-options]
index-strategy = "unsafe-best-match"
extra-index-urls = ["https://download.pytorch.org/whl/cu118"]

[tool.pixi.activation.env]
TRITON_PTXAS_PATH = "$CONDA_PREFIX/lib/python3.10/site-packages/torch/bin/ptxas"

[tool.pixi.tasks]

[tool.black]
line-length = 88
exclude = '''
(
  /(
      \.eggs         # exclude a few common directories in the
    | \.git          # root of the project
    | \.hg
    | \.mypy_cache
    | \.tox
    | \.venv
    | _build
    | buck-out
    | build
    | dist
  )/
)
'''

[tool.isort]
profile = "black"
skip = [
  '.eggs',
  '.git',
  '.hg',
  '.mypy_cache',
  '.nox',
  '.pants.d',
  '.tox',
  '.venv',
  '_build',
  'buck-out',
  'build',
  'dist',
  'node_modules',
  'venv',
]
skip_gitignore = true

[tool.pyright]
typeCheckingMode = "basic"