Skip to content

Commit ab39d74

Browse files
Minor fix on scikit-learn package version for HF spaces. Updated choose_and_run_redactor outputs in cli_redact for successful test run
1 parent e5e86b4 commit ab39d74

File tree

4 files changed

+4
-3
lines changed

4 files changed

+4
-3
lines changed

cli_redact.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1228,6 +1228,7 @@ def main(direct_mode_args={}):
12281228
llm_model_name,
12291229
llm_total_input_tokens,
12301230
llm_total_output_tokens,
1231+
_,
12311232
) = choose_and_run_redactor(
12321233
file_paths=args.input_file,
12331234
prepared_pdf_file_paths=prepared_pdf_paths,

pyproject.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -48,7 +48,7 @@ dependencies = [
4848
"presidio-image-redactor==0.0.57",
4949
"pikepdf==10.3.0",
5050
"pandas==2.3.3",
51-
"scikit-learn==1.8.0",
51+
"scikit-learn<=1.8.0",
5252
"spacy==3.8.11",
5353
"en_core_web_lg @ https://github.com/explosion/spacy-models/releases/download/en_core_web_lg-3.8.0/en_core_web_lg-3.8.0.tar.gz",
5454
"gradio==5.49.1",

requirements.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,7 @@ spaces==0.47.0
3636
awslambdaric==3.1.1
3737

3838
# --- Machine learning / NLP ---
39-
scikit-learn==1.8.0
39+
scikit-learn<=1.8.0
4040
spacy==3.8.11
4141
spaczz==0.6.1
4242
en_core_web_lg @ https://github.com/explosion/spacy-models/releases/download/en_core_web_lg-3.8.0/en_core_web_lg-3.8.0.tar.gz

requirements_lightweight.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@ markdown>=3.7
1515
tabulate>=0.9.0
1616

1717
# --- Machine learning / NLP ---
18-
scikit-learn==1.8.0
18+
scikit-learn<=1.8.0
1919
spacy==3.8.11
2020
spaczz==0.6.1
2121
en_core_web_lg @ https://github.com/explosion/spacy-models/releases/download/en_core_web_lg-3.8.0/en_core_web_lg-3.8.0.tar.gz

0 commit comments

Comments
 (0)