Skip to content

Commit

Permalink
changed names of functions that export the graphml to export_graphml
Browse files Browse the repository at this point in the history
  • Loading branch information
leahh committed Sep 10, 2024
1 parent 27f1ad3 commit 8d0672b
Show file tree
Hide file tree
Showing 5 changed files with 7 additions and 7 deletions.
2 changes: 1 addition & 1 deletion beeflow/common/gdb/gdb_driver.py
Original file line number Diff line number Diff line change
Expand Up @@ -289,5 +289,5 @@ def close(self):
"""Close the connection to the graph database."""

@abstractmethod
def export_dag(self):
def export_graphml(self):
"""Export a BEE workflow as a graphml."""
2 changes: 1 addition & 1 deletion beeflow/common/gdb/neo4j_cypher.py
Original file line number Diff line number Diff line change
Expand Up @@ -727,7 +727,7 @@ def cleanup(tx):
tx.run(cleanup_query)


def export_dag(tx, wf_id):
def export_graphml(tx, wf_id):
"""Export BEE workflow as graphml."""
short_id = wf_id[:6]
export_query = (
Expand Down
4 changes: 2 additions & 2 deletions beeflow/common/gdb/neo4j_driver.py
Original file line number Diff line number Diff line change
Expand Up @@ -472,10 +472,10 @@ def _write_transaction(self, tx_fun, **kwargs):
with self._driver.session() as session:
session.write_transaction(tx_fun, **kwargs)

def export_dag(self, workflow_id):
def export_graphml(self, workflow_id):
"""Export a BEE workflow as a graphml."""
with self._driver.session() as session:
session.write_transaction(tx.export_dag, wf_id=workflow_id)
session.write_transaction(tx.export_graphml, wf_id=workflow_id)


def _reconstruct_requirements(req_records):
Expand Down
4 changes: 2 additions & 2 deletions beeflow/common/wf_interface.py
Original file line number Diff line number Diff line change
Expand Up @@ -305,6 +305,6 @@ def workflow_completed(self):
"""
return self._gdb_driver.workflow_completed(self._workflow_id)

def export_dag(self):
def export_graphml(self):
"""Export a BEE workflow as a graphml."""
self._gdb_driver.export_dag(self._workflow_id)
self._gdb_driver.export_graphml(self._workflow_id)
2 changes: 1 addition & 1 deletion beeflow/wf_manager/resources/wf_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -300,7 +300,7 @@ def setup_workflow(wf_id, wf_name, wf_dir, wf_workdir, no_start, workflow=None,
def export_dag(wf_id):
"""Export the DAG of the workflow."""
wfi = get_workflow_interface(wf_id)
wfi.export_dag()
wfi.export_graphml()


def start_workflow(wf_id):
Expand Down

0 comments on commit 8d0672b

Please sign in to comment.