Fix tensorflow#333 correct typo in memory_store to_dataframe.

Lorenzobattistela · May 31, 2023 · 343661f · 343661f
1 parent 700f982
commit 343661f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tensorflow_similarity/stores/memory_store.py b/tensorflow_similarity/stores/memory_store.py
@@ -201,7 +201,7 @@ def to_data_frame(self, num_records: int = 0) -> PandasDataFrame:
         data = {
             "embeddings": self.embeddings[:num_records],
             "data": self.data[:num_records],
-            "lables": self.labels[:num_records],
+            "labels": self.labels[:num_records],
         }
 
         # forcing type from Any to PandasFrame