diff --git a/environment.yml b/environment.yml
index d1ff29f..373ccaf 100644
--- a/environment.yml
+++ b/environment.yml
@@ -16,3 +16,4 @@ dependencies:
       - psutil
       - pytest
       - matplotlib
+      - memory-profiler
diff --git a/requirements.txt b/requirements.txt
index aa9b013..75bbbe8 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -6,4 +6,5 @@ scikit-learn
 scipy
 psutil
 pytest
-matplotlib
\ No newline at end of file
+matplotlib
+memory-profiler
\ No newline at end of file