Fixed tests

x-tabdeveloping · x-tabdeveloping · commit 64bd30af12f8 · 2026-04-02T14:22:35.000+02:00
diff --git a/tests/test_integration.py b/tests/test_integration.py
@@ -24,7 +24,7 @@
 )
 from turftopic.late import LateSentenceTransformer
 
-ENCODER = "sentence-transformers/static-retrieval-mrl-en-v1"
+ENCODER = "sentence-transformers/paraphrase-MiniLM-L3-v2"
 
 
 def batched(iterable, n: int):
@@ -58,7 +58,7 @@ def generate_dates(
     ],
     remove=("headers", "footers", "quotes"),
 )
-texts = newsgroups.data
+texts = newsgroups.data[:400]
 trf = LateSentenceTransformer(ENCODER)
 embeddings = np.asarray(trf.encode(texts))
 timestamps = generate_dates(n_dates=len(texts))

Original file line number	Diff line number	Diff line change
`@@ -24,7 +24,7 @@`
`24`	`24`	`)`
`25`	`25`	`from turftopic.late import LateSentenceTransformer`
`26`	`26`
`27`		`-ENCODER = "sentence-transformers/static-retrieval-mrl-en-v1"`
	`27`	`+ENCODER = "sentence-transformers/paraphrase-MiniLM-L3-v2"`
`28`	`28`
`29`	`29`
`30`	`30`	`def batched(iterable, n: int):`
`@@ -58,7 +58,7 @@ def generate_dates(`
`58`	`58`	`],`
`59`	`59`	`remove=("headers", "footers", "quotes"),`
`60`	`60`	`)`
`61`		`-texts = newsgroups.data`
	`61`	`+texts = newsgroups.data[:400]`
`62`	`62`	`trf = LateSentenceTransformer(ENCODER)`
`63`	`63`	`embeddings = np.asarray(trf.encode(texts))`
`64`	`64`	`timestamps = generate_dates(n_dates=len(texts))`