From 1ab6b096a2766e795915483bd17b9041a4a46bd6 Mon Sep 17 00:00:00 2001
From: Andras Schmelczer <andras@schmelczer.dev>
Date: Sun, 26 Jun 2022 12:22:31 +0200
Subject: [PATCH] Minor fixes

---
 .github/workflows/{check.yml => test.yml} | 2 +-
 .vscode/tasks.json                        | 4 ++--
 docs/README.md                            | 2 +-
 setup.cfg                                 | 2 +-
 src/great_ai/utilities/__init__.py        | 1 -
 5 files changed, 5 insertions(+), 6 deletions(-)
 rename .github/workflows/{check.yml => test.yml} (99%)

diff --git a/.github/workflows/check.yml b/.github/workflows/test.yml
similarity index 99%
rename from .github/workflows/check.yml
rename to .github/workflows/test.yml
index 4daf861..4279e47 100644
--- a/.github/workflows/check.yml
+++ b/.github/workflows/test.yml
@@ -1,4 +1,4 @@
-name: Check library
+name: Test
 
 on:
   workflow_dispatch:
diff --git a/.vscode/tasks.json b/.vscode/tasks.json
index 9d64445..3ad7159 100644
--- a/.vscode/tasks.json
+++ b/.vscode/tasks.json
@@ -2,7 +2,7 @@
   "version": "2.0.0",
   "tasks": [
     {
-      "label": "Format and lint Python",
+      "label": "Format and lint",
       "type": "shell",
       "command": "source .env/bin/activate && scripts/format-python.sh .",
       "windows": {
@@ -18,7 +18,7 @@
       }
     },
     {
-      "label": "Test Python",
+      "label": "Test",
       "type": "shell",
       "command": "source .env/bin/activate && python3 -m pytest .",
       "windows": {
diff --git a/docs/README.md b/docs/README.md
index 44b0b3d..2ef2e77 100644
--- a/docs/README.md
+++ b/docs/README.md
@@ -3,9 +3,9 @@
 ## Exports
 
 - [clean](src/sus/clean.py)
+- [language](src/sus/clean.py)
 - [unique](src/sus/unique.py)
 - [parallel_map](src/sus/parallel_map.py)
-- [lemmatize](src/sus/lemmatize.py)
 - [evaluate_ranking](src/sus/evaluate_ranking/evaluate_ranking.py)
 - [get_sentences](src/sus/get_sentences.py)
 
diff --git a/setup.cfg b/setup.cfg
index 7af40a1..94ec698 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -23,7 +23,7 @@ install_requires =
     unidecode >= 1.3.0
     multiprocess >= 0.70.0.0
     tqdm >= 4.0.0
-    scikit-learn == 1.1.1
+    scikit-learn >= 1.0.0
     matplotlib >= 3.5.0
     numpy >= 1.22.0
     syntok >= 1.4.0
diff --git a/src/great_ai/utilities/__init__.py b/src/great_ai/utilities/__init__.py
index 7d63c08..c06e891 100644
--- a/src/great_ai/utilities/__init__.py
+++ b/src/great_ai/utilities/__init__.py
@@ -3,7 +3,6 @@ from .config_file import ConfigFile, ParseError
 from .evaluate_ranking import evaluate_ranking
 from .get_sentences import get_sentences
 from .language import english_name_of_language, is_english, predict_language
-from .lemmatize import lemmatize
 from .logger import get_logger
 from .match_names import match_names
 from .parallel_map import parallel_map