diff --git a/environment.yml b/environment.yml
index 6b7f6ba..039ab46 100644
--- a/environment.yml
+++ b/environment.yml
@@ -26,6 +26,7 @@ dependencies:
     - nilearn==0.10.1
     - numpyencoder==0.3.0
     - protobuf==3.20.*
+    - pycaret==3.0.4
     - ray==2.5.1
     - scikit_image==0.18.2
     - SimpleITK==2.1.1.2
diff --git a/pyproject.toml b/pyproject.toml
index 80537f6..91a0aea 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -29,6 +29,7 @@ numpyencoder = "0.3.0"
 pandas = "1.5.3"
 Pillow = "9.2.0"
 protobuf = "3.20.*"
+pycaret = "3.0.4"
 pydicom = "1.2.2"
 PyWavelets = "1.1.1"
 ray = "2.5.1"
diff --git a/requirements.txt b/requirements.txt
index e5521a7..387d576 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -12,6 +12,7 @@ numpyencoder==0.3.0
 pandas==1.5.3
 Pillow==9.2.0
 protobuf==3.20.*
+pycaret==3.0.4
 pydicom==1.2.2
 PyWavelets==1.1.1
 ray==2.5.1