openfold-3/pyproject.toml at 6f60c24d5905bbabbb2774ea92843d8e16c097f5 · aqlaboratory/openfold-3 · 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
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
[build-system]
# torch used at build needs to match the one at runtime
requires = ["setuptools", "wheel"]
build-backend = "setuptools.build_meta"

[project]
name = "openfold3"
version = "0.3.1"
dependencies = [
  "awscli",
  "torch",# ==2.5.1",
  "pytorch-lightning >=2.1",
  "numpy",
  "pandas",
  "PyYAML",
  "requests",
  "scipy",
  "tqdm",
  "typing-extensions",
  "wandb",
  "ml-collections",
  "mkl",
  "aria2",
  "biotite >=1.1.0, <1.3",
  "rdkit",
  "boto3",
  "memory_profiler",
  "func_timeout",
  "lmdb", # == 1.6  # This verison might be required to avoid Error during training
  "ijson",
  "deepspeed",
  "pdbeccdutils",
  # For cutlass >=4, import library is cutlass_cppgen, but deepspeed is
  # currently using cutlass.
  "nvidia-cutlass <4",

  # additional pins

  # Pinned cutlass <4 calls from cuda.__version__, which was removed when
  # cuda-python went fully metapackage
  "cuda-python <12.9.1",
]
authors = [
  {name = "OpenFold Team", email = "support@openfold.io" }
]
description = "OpenFold's Biological Structure Prediction Model based on DeepMind's AlphaFold 3"
readme = "README.md"
requires-python = ">=3.10"
license = { text = "Apache-2.0" }
classifiers = [
    "License :: OSI Approved :: Apache Software License",
    "Operating System :: POSIX :: Linux",
    "Programming Language :: Python :: 3.10",
    "Topic :: Scientific/Engineering :: Artificial Intelligence",
]

[project.urls]
Repository = "https://github.com/aqlaboratory/openfold-3"

[project.scripts]
run_openfold="openfold3.run_openfold:cli"
setup_openfold="openfold3.setup_openfold:main"

[dependency-groups]
test = [
  "pytest",
  "ruff",
  "awscli",
  "aria2",
  "pytest-xdist",
  "pytest-cov",
  "pytest-benchmark",
]

[project.optional-dependencies]
dev = [
  "pytest",
  "ruff",
  # used for documentation
  "sphinx",
  "myst-parser",
  "furo"
]
cuequivariance = [
  "cuequivariance>=0.6.1",
  "cuequivariance-ops-torch-cu12>=0.6.1",
  "cuequivariance-torch>=0.6.1",
  "torch>=2.7",
]

[tool.setuptools]
include-package-data = true

[tool.setuptools.packages.find]
where = ["."]
include = ["openfold3*"]

[tool.setuptools.package-data]
openfold3 = [
  "projects/*/config/*.yml",
  "tests/test_data/**/*",
  "scripts/*.sh"
]

# much of the cibuildwheel config taken from benfred/implicit
[tool.cibuildwheel]
before-all = "devtools/install_cuda.sh"
manylinux-x86_64-image = "manylinux2014"

[tool.cibuildwheel.linux]
repair-wheel-command = """
AUDITWHEEL_PLAT=manylinux2014_$(uname -m) \
auditwheel repair \
    --exclude libc10.so \
    --exclude libtorch.so \
    --exclude libtorch_cpu.so \
    --exclude libtorch_python.so \
    -w {dest_dir} {wheel}
"""

[tool.cibuildwheel.linux.environment]
CUDACXX = "/usr/local/cuda/bin/nvcc"

[tool.ruff]
line-length = 88

[tool.ruff.lint]
select = [
    # pycodestyle
    "E",
    # Pyflakes
    "F",
    # pyupgrade
    "UP",
    # flake8-bugbear
    "B",
    # flake8-simplify
    "SIM",
    # isort
    "I",
]
ignore = [
    "E741",
    "SIM108",
    "B905",  # explicit strict= for zip
]
extend-safe-fixes = ["UP006"]

[tool.pytest.ini_options]
log_cli = true
log_cli_level = "WARNING"
log_cli_format = "%(asctime)s [%(levelname)8s] %(message)s (%(filename)s:%(lineno)s)"
log_cli_date_format = "%Y-%m-%d %H:%M:%S"
testpaths = ["openfold3/tests"]
python_files = ["test_*.py"]
markers = [
    "inference_verification: Tests to verify installation on inference example",
    "slow: Large size test"
]