scikit-learn-contrib · anhkhoangoho · Oct 9, 2023 · Oct 9, 2023 · Oct 9, 2023 · Oct 10, 2023
diff --git a/.gitignore b/.gitignore
@@ -58,6 +58,7 @@ qolmat/.converge
 examples/*.ipynb
 examples/figures/*
 examples/data/*
+!examples/data/imp_pred/
 examples/local
 
 
@@ -66,3 +67,4 @@ examples/local
 
 # Logs
 nohup.txt
+examples/mlruns/*
diff --git a/environment.ci.yml b/environment.ci.yml
@@ -6,6 +6,7 @@ dependencies:
     - codecov
     - flake8
     - matplotlib
+    - plotly
     - mypy
     - numpy
     - numpydoc

diff --git a/environment.dev.yml b/environment.dev.yml
@@ -34,4 +34,8 @@ dependencies:
           - pytest-cov==4.0.0
           - pytest-mock==3.10.0
           - sphinx_markdown_tables==0.0.17
+          - plotly
+          - xgboost
+          - torch
+          - datasets
           - -e .