Use braces

VirusTotal · Jul 28, 2022 · cf50f07 · cf50f07
1 parent 063cf5e
commit cf50f07
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/vt_graph_api/graph.py b/vt_graph_api/graph.py
@@ -458,10 +458,10 @@ def _add_group_nodes_from_json_data(self, json_graph_data_nodes):
       }
     """
 
-    self.group_nodes = dict([
-        (node['entity_id'], node) for node in json_graph_data_nodes
-        if node.get("type") == "relationship" and node.get(
-            "entity_attributes", {}).get("relationship_type") == "group"])
+    self.group_nodes = {node['entity_id']: node for node in
+                        json_graph_data_nodes
+                        if node.get("type") == "relationship" and node.get(
+            "entity_attributes", {}).get("relationship_type") == "group"}
 
   def _is_special_relationship_node(self, node):
     """Checks if a node is a special relationship node.