CaDRe/requirements.txt at main · MinghaoFu/CaDRe · 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
# Runtime dependencies for the CaDRe reference implementation.
# Pinned to the versions used for the ICML 2026 camera-ready experiments.

# Deep learning
torch==2.2.0
torchvision==0.17.0
torchsummary==1.5.1
lightning==2.2.5
pytorch_lightning==2.2.0.post0

# Numerics / data
numpy==1.26.4
scipy==1.13.1
pandas==1.5.3
scikit_learn==1.4.0
h5py==3.10.0
xarray==2024.1.1

# Causal discovery / nonlinear ICA
lingam==1.8.2
python_igraph==0.11.3
networkx==3.2.1
einops==0.8.0
leap==2021.1

# Climate / geospatial
Cartopy==0.22.0

# Imaging
matplotlib==3.8.3
seaborn==0.13.2
imageio==2.34.0
Pillow==10.3.0
opencv_python==4.9.0.80

# RL / simulation (used by some legacy experiments)
gym==0.26.2
pymunk==6.8.0
xvfbwrapper==0.2.9

# Utilities
PyYAML==6.0.1
tqdm==4.66.2
typing_extensions==4.12.1
ipython==8.21.0
ipywidgets==8.1.2

# Test
pytest==8.0.0