From 4ad0722fb566e999235cfcec8fe93975cf801bd8 Mon Sep 17 00:00:00 2001
From: Makris Spiros <makris.spiros@gmail.com>
Date: Fri, 5 May 2023 09:23:27 +0300
Subject: [PATCH] fix typo in SentenceEmbeddingsTokenizerOuput

---
 src/pipelines/sentence_embeddings/mod.rs      | 2 +-
 src/pipelines/sentence_embeddings/pipeline.rs | 8 ++++----
 2 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/src/pipelines/sentence_embeddings/mod.rs b/src/pipelines/sentence_embeddings/mod.rs
index 974c33de..2c4f275b 100644
--- a/src/pipelines/sentence_embeddings/mod.rs
+++ b/src/pipelines/sentence_embeddings/mod.rs
@@ -42,7 +42,7 @@ pub use config::{
 };
 pub use pipeline::{
     SentenceEmbeddingsModel, SentenceEmbeddingsModelOutput, SentenceEmbeddingsOption,
-    SentenceEmbeddingsTokenizerOuput,
+    SentenceEmbeddingsTokenizerOutput,
 };
 
 pub use resources::{
diff --git a/src/pipelines/sentence_embeddings/pipeline.rs b/src/pipelines/sentence_embeddings/pipeline.rs
index 24915185..c5b148d1 100644
--- a/src/pipelines/sentence_embeddings/pipeline.rs
+++ b/src/pipelines/sentence_embeddings/pipeline.rs
@@ -283,7 +283,7 @@ impl SentenceEmbeddingsModel {
     }
 
     /// Tokenizes the inputs
-    pub fn tokenize<S>(&self, inputs: &[S]) -> SentenceEmbeddingsTokenizerOuput
+    pub fn tokenize<S>(&self, inputs: &[S]) -> SentenceEmbeddingsTokenizerOutput
     where
         S: AsRef<str> + Sync,
     {
@@ -327,7 +327,7 @@ impl SentenceEmbeddingsModel {
             .map(|input| Tensor::of_slice(&(input)))
             .collect::<Vec<_>>();
 
-        SentenceEmbeddingsTokenizerOuput {
+        SentenceEmbeddingsTokenizerOutput {
             tokens_ids,
             tokens_masks,
         }
@@ -341,7 +341,7 @@ impl SentenceEmbeddingsModel {
     where
         S: AsRef<str> + Sync,
     {
-        let SentenceEmbeddingsTokenizerOuput {
+        let SentenceEmbeddingsTokenizerOutput {
             tokens_ids,
             tokens_masks,
         } = self.tokenize(inputs);
@@ -457,7 +457,7 @@ impl SentenceEmbeddingsModel {
 }
 
 /// Container for the SentenceEmbeddings tokenizer output.
-pub struct SentenceEmbeddingsTokenizerOuput {
+pub struct SentenceEmbeddingsTokenizerOutput {
     pub tokens_ids: Vec<Tensor>,
     pub tokens_masks: Vec<Tensor>,
 }