Update environment.yml

Signed-off-by: Johannes Brøns Christensen <j.brons.christensen@gmail.com>
JohannesBroens · Dec 22, 2023 · e860e90 · e860e90
1 parent 1d13e89
commit e860e90
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/.github/workflows/environment.yml b/.github/workflows/environment.yml
@@ -2,17 +2,16 @@ name: ML_QC
 channels:
   - pytorch
   - nvidia
-  - scikit-learn
   - conda-forge
   - defaults
 dependencies:
-  - python=3.10
+  - python
   - pytorch
   - pytorch-cuda
   - py-xgboost-gpu
   - numpy
   - seaborn
-  - sklearn
+  - scikit-learn
   - codecov
   - jupyter
   - ipython3