earth2studio/tox-smoke.ini at main · NVIDIA/earth2studio · 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
[tox]
requires =
    tox >= 4.0
    tox-uv >= 1.28.0
envlist = {py311,py312,py313}

[testenv]
package = skip # Important for smoke tests! We will install e2s manually!
passenv =
    EARTH2STUDIO_CACHE
    EARTH2STUDIO_PACKAGE_TIMEOUT
    UV_CACHE_DIR
    VERSION
    ARTIFACTORY_USER
    ARTIFACTORY_API_TOKEN
    ARTIFACTORY_URL
    NGC_CLI_API_KEY
    S3FS_ENDPOINT
    S3FS_KEY
    S3FS_SECRET
    HF_TOKEN
    CDSAPI_KEY
    CDSAPI_URL
    MAX_JOBS
setenv =
    # https://docs.jax.dev/en/latest/gpu_memory_allocation.html
    XLA_PYTHON_CLIENT_ALLOCATOR = platform
    XLA_PYTHON_CLIENT_PREALLOCATE = false
    # https://github.com/Dao-AILab/flash-attention/issues/2008
    FLASH_ATTENTION_FORCE_BUILD = TRUE
    FLASH_ATTENTION_FORCE_CXX11_ABI = FALSE
    FLASH_ATTENTION_SKIP_CUDA_BUILD = FALSE
    FLASH_ATTN_CUDA_ARCHS = "90;120"
runner = uv-venv-lock-runner
description = Run smoke tests of wheel through python version sweep
commands =
    uv sync
    uv sync --extra all
    uv pip install earth2studio=={env:VERSION} \
      --extra-index-url="https://{env:ARTIFACTORY_USER}:{env:ARTIFACTORY_API_TOKEN}@{env:ARTIFACTORY_URL}"
    pytest {posargs:--no-testmon} test/