From 52036c10d7726aef51220e3e557e5c29d482f41a Mon Sep 17 00:00:00 2001
From: Peilin Yu <54282945+dotpyu@users.noreply.github.com>
Date: Fri, 5 Jul 2024 14:47:39 -0400
Subject: [PATCH] Update grpc.py

Fixing logit typo -> logits
---
 alfred/fm/remote/grpc.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/alfred/fm/remote/grpc.py b/alfred/fm/remote/grpc.py
index 4f4b179..6a7aafe 100644
--- a/alfred/fm/remote/grpc.py
+++ b/alfred/fm/remote/grpc.py
@@ -136,7 +136,7 @@ def _run_req_gen():
                         **{
                             "prediction": response.message,
                             "scores": scores,
-                            "logit": logits,
+                            "logits": logits,
                             "embeddings": bytes_to_tensor(response.embedding),
                         }
                     )