Update discourse_graph.py

discourse_graph.py

@@ -2,12 +2,10 @@ import networkx as nx
 
 DG = nx.DiGraph()
 
-
 def add_claim(company, claim, evidence):
 
-    DG.add_node(claim, type="claim")
-
-    DG.add_node(evidence, type="evidence")
+    DG.add_node(claim)
+    DG.add_node(evidence)
 
     DG.add_edge(evidence, claim, relation="supports")
 
@@ -23,7 +21,7 @@ def detect_greenwashing():
 
     for u,v,d in DG.edges(data=True):
 
-        if d.get("relation") == "contradicts":
+        if d["relation"] == "contradicts":
 
             issues.append((u,v))