Remove scikit-learn dependency (#347)

sdv-dev · Mar 20, 2024 · bc465e6 · bc465e6
1 parent d861c4a
commit bc465e6
Show file tree

Hide file tree

Showing 3 changed files with 1 addition and 5 deletions.
diff --git a/Makefile b/Makefile
@@ -235,5 +235,5 @@ release-major: check-release bumpversion-major release
 
 .PHONY: check-deps
 check-deps:
-	$(eval allow_list='numpy=|pandas=|scikit-learn=|tqdm=|torch=|rdt=')
+	$(eval allow_list='numpy=|pandas=|tqdm=|torch=|rdt=')
 	pip freeze | grep -v "CTGAN.git" | grep -E $(allow_list) > $(OUTPUT_FILEPATH)
diff --git a/latest_requirements.txt b/latest_requirements.txt
@@ -1,6 +1,5 @@
 numpy==1.26.4
 pandas==2.2.1
 rdt==1.10.0
-scikit-learn==1.4.1.post1
 torch==2.2.1
 tqdm==4.66.2
diff --git a/pyproject.toml b/pyproject.toml
@@ -25,8 +25,6 @@ dependencies = [
     "pandas>=1.1.3;python_version<'3.10'",
     "pandas>=1.3.4;python_version>='3.10' and python_version<'3.11'",
     "pandas>=1.5.0;python_version>='3.11'",
-    "scikit-learn>=0.24;python_version<'3.10'",
-    "scikit-learn>=1.1.3;python_version>='3.10'",
     "torch>=1.8.0;python_version<'3.10'",
     "torch>=1.11.0;python_version>='3.10' and python_version<'3.11'",
     "torch>=2.0.0;python_version>='3.11'",
@@ -123,7 +121,6 @@ namespaces = false
     '*.png',
     '*.gif'
 ]
-'tests' = ['*']
 
 [tool.setuptools.exclude-package-data]
 '*' = [