From 3f6d2d879a12300e7ce2a973dbf5aa76b0f2a491 Mon Sep 17 00:00:00 2001 From: Cyber MacGeddon Date: Tue, 23 Jul 2024 21:08:21 +0100 Subject: [PATCH] Fix more naming problems --- trustgraph/kg/extract_relationships/extract.py | 2 +- trustgraph/storage/graph_embeddings/milvus/write.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/trustgraph/kg/extract_relationships/extract.py b/trustgraph/kg/extract_relationships/extract.py index 3ef180d5..45f63e8a 100755 --- a/trustgraph/kg/extract_relationships/extract.py +++ b/trustgraph/kg/extract_relationships/extract.py @@ -87,7 +87,7 @@ class Processor(ConsumerProducer): def emit_vec(self, ent, vec): - r = VectorsAssociation(entity=ent, vectors=vec) + r = GraphEmbeddings(entity=ent, vectors=vec) self.vec_prod.send(r) def handle(self, msg): diff --git a/trustgraph/storage/graph_embeddings/milvus/write.py b/trustgraph/storage/graph_embeddings/milvus/write.py index 1f7a3f15..1a6883a1 100755 --- a/trustgraph/storage/graph_embeddings/milvus/write.py +++ b/trustgraph/storage/graph_embeddings/milvus/write.py @@ -3,7 +3,7 @@ Accepts entity/vector pairs and writes them to a Milvus store. """ -from .... schema import VectorsAssociation +from .... schema import GraphEmbeddings from .... schema import graph_embeddings_store_queue from .... log_level import LogLevel from .... triple_vectors import TripleVectors @@ -27,7 +27,7 @@ class Processor(Consumer): **params | { "input_queue": input_queue, "subscriber": subscriber, - "input_schema": VectorsAssociation, + "input_schema": GraphEmbeddings, "store_uri": store_uri, } )