From f67b051c84b8ca343ba1c749815867f106a0dc7a Mon Sep 17 00:00:00 2001
From: James Briggs <35938317+jamescalam@users.noreply.github.com>
Date: Sun, 7 Jan 2024 16:19:17 +0100
Subject: [PATCH] add type definition for self.encoder
---
semantic_router/layer.py | 6 ++----
1 file changed, 2 insertions(+), 4 deletions(-)
diff --git a/semantic_router/layer.py b/semantic_router/layer.py
index dd9006ca..08261756 100644
--- a/semantic_router/layer.py
+++ b/semantic_router/layer.py
@@ -4,10 +4,7 @@ import os
import numpy as np
import yaml
-from semantic_router.encoders import (
- BaseEncoder,
- OpenAIEncoder
-)
+from semantic_router.encoders import BaseEncoder, OpenAIEncoder
from semantic_router.linear import similarity_matrix, top_scores
from semantic_router.llms import BaseLLM, OpenAILLM
from semantic_router.route import Route
@@ -153,6 +150,7 @@ class RouteLayer:
index: np.ndarray | None = None
categories: np.ndarray | None = None
score_threshold: float
+ encoder: BaseEncoder
def __init__(
self,
--
GitLab