Update knowledge_level_and_agent_type.py

RTXteam · May 15, 2024 · 1dc1773 · 1dc1773
1 parent e5ba8e5
commit 1dc1773
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/knowledge_level_and_agent_type.py b/knowledge_level_and_agent_type.py
@@ -57,6 +57,8 @@ def process_edges(input_edges_file, output_file_location, output_file):
         edge_dict['knowledge_level'] = 'not_provided'
         edges_info.write(edge_dict)
 
+    kg2_util.end_read_jsonlines(edges_read_jsonlines_data)
+
 
 if __name__ == '__main__':
     args = make_arg_parser().parse_args()