mirror of
https://github.com/neuml/txtai.git
synced 2026-10-12 02:19:52 +00:00
test: isolate uncentered LEMUR coverage
This commit is contained in:
@@ -272,7 +272,7 @@ class TestPooling(unittest.TestCase):
|
||||
seed=42,
|
||||
)
|
||||
|
||||
pooling = PoolingFactory.create({"path": model, "device": self.device, "modelargs": {"lemur": {"path": output}}})
|
||||
pooling = PoolingFactory.create({"path": model, "device": self.device, "modelargs": {"lemur": {"path": output}, "center": False}})
|
||||
texts = ["Short text.", "A considerably longer synthetic document exercises padding behavior."]
|
||||
queries = pooling.encode(texts, category="query")
|
||||
documents = pooling.encode(texts, category="data")
|
||||
|
||||
Reference in New Issue
Block a user