From 8a2126bba5ddeb09832934d3473e36bb26eae703 Mon Sep 17 00:00:00 2001
From: cybermaggedon <cybermaggedon@gmail.com>
Date: Sat, 2 Nov 2024 10:51:26 +0000
Subject: [PATCH] Fix neo4j schema - incompatible with extractors (#135)

- Fixed schema invocation on Neo4j writer
- Fixed confusing typo name in cassandra template
- Unnecessary import on Cassandra removed
---
 templates/components/cassandra.jsonnet                        | 2 +-
 trustgraph-flow/trustgraph/storage/triples/cassandra/write.py | 2 +-
 trustgraph-flow/trustgraph/storage/triples/neo4j/write.py     | 4 ++--
 3 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/templates/components/cassandra.jsonnet b/templates/components/cassandra.jsonnet
index b9345fed..b52d4b04 100644
--- a/templates/components/cassandra.jsonnet
+++ b/templates/components/cassandra.jsonnet
@@ -24,7 +24,7 @@ cassandra + {
                     .with_reservations("0.1", "128M");
 
             local containerSet = engine.containers(
-                "stop-triples", [ container ]
+                "store-triples", [ container ]
             );
 
             local service =
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,
             }
         )