Make internal method private

Author: oskarhane

src/neo4j_graphrag/experimental/pipeline/pipeline.py

@@ -201,15 +201,15 @@ def draw(
         self, path: str, layout: str = "dot", hide_unused_outputs: bool = True
     ) -> Any:
         """Render the pipeline graph to an HTML file at the specified path"""
-        G = self.get_neo4j_viz_graph(hide_unused_outputs)
+        G = self._get_neo4j_viz_graph(hide_unused_outputs)
 
         # Write the visualization to an HTML file
         with open(path, "w") as f:
             f.write(G.render().data)
 
         return G
 
-    def get_neo4j_viz_graph(
+    def _get_neo4j_viz_graph(
         self, hide_unused_outputs: bool = True
     ) -> VisualizationGraph:
         """Generate a neo4j-viz visualization of the pipeline graph"""
@@ -321,11 +321,11 @@ def get_pygraphviz_graph(self, hide_unused_outputs: bool = True) -> Any:
         Uses neo4j-viz instead of pygraphviz.
         """
         warnings.warn(
-            "get_pygraphviz_graph is deprecated, use get_neo4j_viz_graph instead",
+            "get_pygraphviz_graph is deprecated, use draw instead",
             DeprecationWarning,
             stacklevel=2,
         )
-        return self.get_neo4j_viz_graph(hide_unused_outputs)
+        return self._get_neo4j_viz_graph(hide_unused_outputs)
 
     def add_component(self, component: Component, name: str) -> None:
         """Add a new component. Components are uniquely identified

tests/unit/experimental/pipeline/test_pipeline.py

@@ -387,12 +387,12 @@ def test_pipeline_to_viz() -> None:
     pipe.add_component(component_a, "a")
     pipe.add_component(component_b, "b")
     pipe.connect("a", "b", {"number1": "a.result"})
-    g = pipe.get_neo4j_viz_graph()
+    g = pipe._get_neo4j_viz_graph()
     # 3 nodes:
     #   - 2 components 'a' and 'b'
     #   - 1 output 'a.result'
     assert len(g.nodes) == 3
-    g = pipe.get_neo4j_viz_graph(hide_unused_outputs=False)
+    g = pipe._get_neo4j_viz_graph(hide_unused_outputs=False)
     # 4 nodes:
     #   - 2 components 'a' and 'b'
     #   - 2 output 'a.result' and 'b.result'