research/environment.yml at main · dseinternational/research · 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
name: dse-research
channels:
  - conda-forge
dependencies:
  # ── shared conda-forge compiled core ──────────────────────────────────────
  # Single source of truth: src/python/src/dse_research_utils/data/environment-core.yml
  # These lines must match that spec exactly. Verify with:
  #   dse-check-env environment.yml
  - python=3.14.6
  - numpy>=2.4.6,<2.5 # pymc -> pytensor 3.x -> numba<=0.65.1 -> numpy<2.5
  - scipy>=1.18.0
  - pandas>=3.0.3
  - pyarrow>=24.0.0 # pandas 3.x's default `str` dtype is Arrow-backed; floor held at 24 by jaxlib's libabseil pin — see environment-core.yml
  - numba>=0.58,<=0.65.1 # every pytensor 3.x caps numba here (verified 3.1.3, 3.2.1, 3.2.2); this is what pymc commits us to
  - matplotlib>=3.11.0
  - scikit-learn>=1.9.0
  - statsmodels>=0.14.6
  - pymc>=6.1.0 # pulls pytensor + a C toolchain from conda-forge; each pymc 6.x pins one pytensor 3.x minor (6.1 -> 3.1, 6.2 -> 3.2), which caps numba <=0.65.1 -> numpy <2.5. Floor is the minimum supported, not what solves: 6.2.0/pytensor 3.2.2 is current
  - nutpie>=0.16.11
  - numpyro>=0.21.0
  - jax>=0.10.2 # CPU build; GPU acceleration via environment-gpu.yml
  - arviz>=1.2.0
  - arviz-base>=1.2.0
  - arviz-stats>=1.2.0
  - arviz-plots>=1.2.0
  - preliz>=0.27.0
  - h5py>=3.12.0
  - h5netcdf>=1.3.0
  - graphviz>=14.1.2
  - python-graphviz>=0.20
  # ── research add-ons (compiled / columnar data / system binaries) ─────────
  - lightgbm>=4.6.0
  - xgboost>=3.3.0
  - shap>=0.52.0
  - polars>=1.42.1
  - duckdb>=1.5.4
  - pyreadstat>=1.3.5
  - xarray>=2026.4.0
  - zarr
  - networkx
  - pandoc
  - pip
  - pip:
      # This repo installed editable, with the shared extras. This pulls the
      # library's own runtime deps (azure-identity/-storage-blob, psutil, rich,
      # pyyaml, …) and the pure-Python tail (seaborn, ipython/jupyter, dcor,
      # optuna, orjson/tabulate) via the extras — no need to list them again.
      - -e ./src/python[viz,notebook,dependence,tuning,io]
      # research-only pure-Python
      - formulaic
      - mpmath
      - pingouin
      - plotly
      - pyxlsb
      # typing stubs + dev / build tooling
      - pandas-stubs
      - scipy-stubs
      - hatch
      - build
      - pytest>=9.1.1
      - ruff>=0.15.20
      - setuptools>=82.0.1