Update src/mattext/models/predict.py

Co-authored-by: Kevin M Jablonka <[email protected]>

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