Add pytest

mkranzlein · Aug 20, 2023 · e22332f · e22332f
1 parent 70aa911
commit e22332f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/environment.yml b/environment.yml
@@ -10,6 +10,7 @@ dependencies:
   - pandas
   - pip
   - pyg::pyg
+  - pytest
   - pytorch::pytorch
   - pytorch::pytorch-cuda=11.7
   - scikit-learn

diff --git a/environment_apple_silicon.yml b/environment_apple_silicon.yml
@@ -10,8 +10,9 @@ dependencies:
   - numpy
   - pandas
   - pip
-  - pytorch::pytorch
+  - pytest
   - python
+  - pytorch::pytorch
   - scikit-learn
   - pip:
     - torch_geometric