From ab0578c68ff13bde926db9fe8cedb4fb61102dfa Mon Sep 17 00:00:00 2001
From: Simonas <20096648+simjak@users.noreply.github.com>
Date: Fri, 16 Feb 2024 17:08:26 +0200
Subject: [PATCH] chore: Encoder
---
semantic_router/splitters/rolling_window.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/semantic_router/splitters/rolling_window.py b/semantic_router/splitters/rolling_window.py
index 57bf4733..ad5c568e 100644
--- a/semantic_router/splitters/rolling_window.py
+++ b/semantic_router/splitters/rolling_window.py
@@ -3,6 +3,7 @@ from typing import List
import numpy as np
from matplotlib import pyplot as plt
from nltk.tokenize import word_tokenize
+from semantic_router.encoders.base import BaseEncoder
from semantic_router.schema import DocumentSplit
from semantic_router.splitters.base import BaseSplitter
@@ -35,7 +36,7 @@ class RollingWindowSplitter(BaseSplitter):
def __init__(
self,
- encoder,
+ encoder: BaseEncoder,
score_threshold=0.3,
window_size=5,
plot_splits=False,
--
GitLab