Skip to content

Commit 866fec6

Browse files
jh941213claude
andcommitted
Apply ruff format to kdb_router.py
Co-Authored-By: Claude Opus 5.5 (1M context) <noreply@anthropic.com>
1 parent ab29d59 commit 866fec6

1 file changed

Lines changed: 6 additions & 2 deletions

File tree

‎router_inference/router/kdb_router.py‎

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -38,15 +38,19 @@ def __init__(self, router_name: str):
3838

3939
with open(_CFG) as f:
4040
cfg = json.load(f)
41-
self._embed = SentenceTransformer(cfg["embed_model"], revision=cfg["embed_revision"])
41+
self._embed = SentenceTransformer(
42+
cfg["embed_model"], revision=cfg["embed_revision"]
43+
)
4244
self._classes = cfg["classes"]
4345
self._coef = np.array(cfg["coef"])
4446
self._intercept = np.array(cfg["intercept"])
4547
self._policy = cfg["policy"]
4648
self._default = cfg["default"]
4749

4850
def route_option(self, query: str) -> str:
49-
x = self._embed.encode([routing_text(query)], normalize_embeddings=True, show_progress_bar=False)[0]
51+
x = self._embed.encode(
52+
[routing_text(query)], normalize_embeddings=True, show_progress_bar=False
53+
)[0]
5054
cat = self._classes[int(np.argmax(self._coef @ x + self._intercept))]
5155
return self._policy.get(cat, self._default)
5256

0 commit comments

Comments
 (0)