Skip to content

Commit 0fa2222

Browse files
author
jkasiraj
committed
fix lint failure and typos
1 parent 2ebbf6b commit 0fa2222

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

src/sagemaker/lineage/query.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -362,7 +362,7 @@ def _add_legend(self, path):
362362
with open(path, "w", encoding="utf8") as file:
363363
file.write(html)
364364

365-
def render(self, elements, path="lineage_graph_pyvis.html.html"):
365+
def render(self, elements, path="lineage_graph_pyvis.html"):
366366
"""Render graph for lineage query result.
367367
368368
Args:
@@ -374,7 +374,8 @@ def render(self, elements, path="lineage_graph_pyvis.html.html"):
374374
elements["edges"] contains list of tuples, each tuple represents an edge
375375
format: (edge source arn, edge destination arn, edge association type)
376376
377-
path(optional): The path/filename of the rendered graph html file. (default path: "lineage_graph_pyvis.html.html")
377+
path(optional): The path/filename of the rendered graph html file.
378+
(default path: "lineage_graph_pyvis.html")
378379
379380
Returns:
380381
display graph: The interactive visualization is presented as a static HTML file.
@@ -515,7 +516,7 @@ def _get_visualization_elements(self):
515516
elements = {"nodes": verts, "edges": edges}
516517
return elements
517518

518-
def visualize(self, path: Optional[str] = "lineage_graph_pyvis.html.html"):
519+
def visualize(self, path: Optional[str] = "lineage_graph_pyvis.html"):
519520
"""Visualize lineage query result.
520521
521522
Creates a PyvisVisualizer object to render network graph with Pyvis library.
@@ -524,7 +525,8 @@ def visualize(self, path: Optional[str] = "lineage_graph_pyvis.html.html"):
524525
PyvisVisualizer for rendering graph.
525526
526527
Args:
527-
path(optional): The path/filemname of the rendered graph html file. (default path: "lineage_graph_pyvis.html.html")
528+
path(optional): The path/filename of the rendered graph html file.
529+
(default path: "lineage_graph_pyvis.html")
528530
529531
Returns:
530532
display graph: The interactive visualization is presented as a static HTML file.

0 commit comments

Comments
 (0)