olmoearth_pretrain_minimal/pyproject.toml at main · allenai/olmoearth_pretrain_minimal · 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
[project]
name = "olmoearth-pretrain-minimal"
version = "0.0.4"
description = "Minimal package for loading and initializing OlmoEarth models"
authors = [
    { name = "OlmoEarth Team" },
]
readme = "README.md"
license = {file = "LICENSE"}
requires-python = ">=3.11"
dependencies = [
    "einops>=0.7.0",
    "huggingface_hub",
    "numpy>=1.26.4",
    "universal-pathlib>=0.2.5",
    "torch>=2.7",
]

[project.optional-dependencies]
dev = [
    "pytest>=7.0.0",
    "olmoearth-pretrain==0.1.0",
]

[build-system]
requires = ["setuptools>=61"]
build-backend = "setuptools.build_meta"

[tool.setuptools.packages.find]
include = ["olmoearth_pretrain_minimal*"]

[tool.setuptools.package-data]
"olmoearth_pretrain_minimal.olmoearth_pretrain_v1.data.norm_configs" = ["*.json"]

[tool.pytest.ini_options]
markers = [
    "slow: marks tests as slow (deselect with '-m \"not slow\"')",
]
testpaths = ["tests"]