From 18fb868dd15d66ea12af8b2340ca3f7e255faccf Mon Sep 17 00:00:00 2001
From: Nawaf <86834161+n0w0f@users.noreply.github.com>
Date: Sun, 2 Jun 2024 17:14:08 +0200
Subject: [PATCH] Update src/mattext/models/predict.py

Co-authored-by: Kevin M Jablonka <32935233+kjappelbaum@users.noreply.github.com>
---
 src/mattext/models/predict.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/mattext/models/predict.py b/src/mattext/models/predict.py
index 5642df3..490d1d4 100644
--- a/src/mattext/models/predict.py
+++ b/src/mattext/models/predict.py
@@ -82,4 +82,4 @@ def predict(self):
         prediction_ids = self.tokenized_test_datasets["mbid"]
         self.prediction_ids = preidction_ids
 
-        return pd.Series(predictions.predictions.flatten()),preidction_ids
+        return pd.Series(predictions.predictions.flatten()),prediction_ids