f

segment-any-text · Jun 16, 2024 · 1add5b5 · 1add5b5
1 parent c995b70
commit 1add5b5
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/wtpsplit/evaluation/llm_sentence.py b/wtpsplit/evaluation/llm_sentence.py
@@ -425,7 +425,7 @@ def align_llm_output(row):
         )
         # same as aligned_in, aligned_llm, but with additional formatting. Latter used to debug only.
         formatted_alignment = alignment._format_alignment(aligned_in, aligned_llm).split("\n")
-    except:
+    except:    # ruff: ignore=E722
         print("Alignment failed: ", row.name)
         formatted_alignment = [row["test_chunks"], "", " " * len(row["test_chunks"])]
     return pd.Series(

diff --git a/wtpsplit/train/train.py b/wtpsplit/train/train.py
@@ -34,7 +34,7 @@
 )
 from wtpsplit.train.evaluate import evaluate_sentence
 from wtpsplit.train.trainer import Trainer
-from wtpsplit.train.utils import Model, cleanup_cache_files
+from wtpsplit.train.utils import Model
 # from wtpsplit.train.utils import cleanup_cache_files
 from wtpsplit.utils import Constants, LabelArgs, corrupt_training, get_label_dict, get_subword_label_dict