From a2cf0c2b5fdaee1e8084a81d06683566de230ad9 Mon Sep 17 00:00:00 2001
From: Danilo Ferreira de Lima <danilo.enoque.ferreira.de.lima@xfel.de>
Date: Fri, 6 Oct 2023 14:32:41 +0200
Subject: [PATCH] Update instructions.

---
 README.md | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/README.md b/README.md
index e168ba9..9af2e50 100644
--- a/README.md
+++ b/README.md
@@ -24,7 +24,7 @@ source env/bin/activate
 pip install --force-reinstall --index-url https://pypi.anaconda.org/intel/simple --no-dependencies numpy scipy==1.7.3
 pip install numpy scipy==1.7.3
 pip install torch --index-url https://download.pytorch.org/whl/cpu
-pip install joblib
+pip install joblib scikit-learn
 
 # install dependencies for the analysis in the notebook
 pip install matplotlib lmfit seaborn extra_data
@@ -50,7 +50,7 @@ pip install --force-reinstall --index-url https://pypi.anaconda.org/intel/simple
 pip install numpy scipy==1.7.3
 # install PyTorch
 pip install torch --index-url https://download.pytorch.org/whl/cpu
-pip install joblib
+pip install joblib scikit-learn
 ```
 
 ## Usage
-- 
GitLab