瀏覽代碼

Fix storage bug

jherve 1 年之前
父節點
當前提交
2d7c272ce7
共有 1 個文件被更改,包括 2 次插入2 次删除
  1. 2 2
      src/de_quoi_parle_le_monde/storage.py

+ 2 - 2
src/de_quoi_parle_le_monde/storage.py

@@ -418,8 +418,8 @@ class Storage:
 
     @classmethod
     def _from_articles_embeddings_row(cls, r):
-        [embeds_view] = [v for v in cls.views if v.name == "articles_embeddings"]
-        d = cls._from_row(r, embeds_view)
+        [embeds_table] = [t for t in cls.tables if t.name == "articles_embeddings"]
+        d = cls._from_row(r, embeds_table)
         d.update(title_embedding=np.frombuffer(d["title_embedding"], dtype="float32"))
 
         return d