Skip to content

Commit

Permalink
test: fix GraphCreator test
Browse files Browse the repository at this point in the history
  • Loading branch information
theissenhelen committed Jul 30, 2024
1 parent 1fbac66 commit 9642666
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tests/test_create.py
Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,7 @@ def test_generate_graph(self, config_file: tuple[Path, str], mock_grids_path: tu
graph_path = tmp_path / name if isinstance(name, str) else None
config_path = tmp_path / config_name

graph = GraphCreator(config=config_path, path=graph_path).create()
graph = GraphCreator(config=config_path).create(save_path=graph_path)

assert isinstance(graph, HeteroData)
assert "test_nodes" in graph.node_types
Expand Down

0 comments on commit 9642666

Please sign in to comment.