-
Notifications
You must be signed in to change notification settings - Fork 31
/
environment.yml
60 lines (60 loc) · 1.85 KB
/
environment.yml
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
name: project-azua
channels:
- pytorch
- defaults
# Do we need conda-forge?
- conda-forge
dependencies:
- python==3.8.2
- tqdm==4.50.2
# ML packages
- numpy==1.21.0
- pytorch==1.7.0
- cudatoolkit=10.2
- scipy==1.5.2
- scikit-learn==0.23.2
- pandas==1.1.3
- matplotlib==3.3.2
- torchvision>=0.8.0
- seaborn==0.11.0
- tensorboard==2.3.0
- Flask==1.1.4
- jupyter
# Dev tools (testing, static analysis, styling etc.):
- pytest==6.1.1
- pytest-cov==2.10.1
- mypy==0.931
- gitpython==3.1.11
- black==19.10b0
- flake8==3.8.4
- pre-commit==2.9.3
- pip==20.2.4
- pip:
# ML packages
- torch-scatter==2.0.5+cu102
- --find-links https://pytorch-geometric.com/whl/torch-1.7.0.html
- torch-sparse==0.6.8+cu102
- --find-links https://pytorch-geometric.com/whl/torch-1.7.0.html
- torch-geometric==1.6.3
- transformers==4.4.2
- sentence_transformers==1.0.4
# Note: we will not need it once numpy 1.20, which contains stubs, is released (and we move to it)
- git+https://github.com/numpy/numpy-stubs
- dppy==0.3.1
# DI framework
- dependency_injector==4.34.2
# gcastle for causality baselines
# Should be updated to the Huawei codebase once they release new full-pytorch version (#21024)
- git+https://github.com/tomsons22/trustworthyAI.git@c2e3535eba28cabf8e1309f55bffc3e4d4d78c00#subdirectory=gcastle
# For float32 (bug #20930)
# - git+https://github.com/tomsons22/trustworthyAI.git@bf6e050b6c2511d857ac687fe8c01fcf5fe32f02#subdirectory=gcastle
# For DoWhy baseline
- graphviz==0.17
# Use DoWhy from Github to allow for faster integration of new features
- git+https://github.com/microsoft/dowhy.git@8b2cf6a
- econml==0.12.0
- networkx[default]==2.6.3
- igraph==0.9.9
# Required by mypy for Library stubs
- types-PyYAML
- numpyro==0.9.1