cleanup: create_graph

arangoml · Dec 27, 2023 · 3ca00eb · 3ca00eb
1 parent 44dd28f
commit 3ca00eb
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/arango_datasets/datasets.py b/arango_datasets/datasets.py
@@ -122,9 +122,8 @@ def load(
                     self.__import_bulk(col, load_function(file))
 
         if edge_definitions := dataset_contents.get("edge_definitions"):
-            graph_name = dataset_contents["graph_name"]
-            self.user_db.delete_graph(graph_name, ignore_missing=True)
-            self.user_db.create_graph(graph_name, edge_definitions)
+            self.user_db.delete_graph(dataset_name, ignore_missing=True)
+            self.user_db.create_graph(dataset_name, edge_definitions)
 
     def __get_response(self, url: str, timeout: int = 60) -> requests.Response:
         """Wrapper around requests.get() with a progress bar.