From 555941d8611512d79d53ef9abaa1bbacbc99f6c8 Mon Sep 17 00:00:00 2001 From: Cyber MacGeddon Date: Sat, 2 Nov 2024 10:47:36 +0000 Subject: [PATCH] - Fixed schema invocation on Neo4j writer - Fixed confusing typo name in cassandra template - Unnecessary import on Cassandra removed --- trustgraph-flow/trustgraph/storage/triples/cassandra/write.py | 2 +- trustgraph-flow/trustgraph/storage/triples/neo4j/write.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/trustgraph-flow/trustgraph/storage/triples/cassandra/write.py b/trustgraph-flow/trustgraph/storage/triples/cassandra/write.py index 6825d452..e7078e08 100755 --- a/trustgraph-flow/trustgraph/storage/triples/cassandra/write.py +++ b/trustgraph-flow/trustgraph/storage/triples/cassandra/write.py @@ -10,7 +10,7 @@ import argparse import time from .... direct.cassandra import TrustGraph -from .... schema import Triple, Triples +from .... schema import Triples from .... schema import triples_store_queue from .... log_level import LogLevel from .... base import Consumer diff --git a/trustgraph-flow/trustgraph/storage/triples/neo4j/write.py b/trustgraph-flow/trustgraph/storage/triples/neo4j/write.py index c346e5c4..68a9085d 100755 --- a/trustgraph-flow/trustgraph/storage/triples/neo4j/write.py +++ b/trustgraph-flow/trustgraph/storage/triples/neo4j/write.py @@ -11,7 +11,7 @@ import time from neo4j import GraphDatabase -from .... schema import Triple +from .... schema import Triples from .... schema import triples_store_queue from .... log_level import LogLevel from .... base import Consumer @@ -39,7 +39,7 @@ class Processor(Consumer): **params | { "input_queue": input_queue, "subscriber": subscriber, - "input_schema": Triple, + "input_schema": Triples, "graph_host": graph_host, } )