Use pygraph even less

seddonym · Dec 9, 2024 · ccd280a · ccd280a
1 parent 551ce74
commit ccd280a
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/src/grimp/adaptors/rustgraph.py b/src/grimp/adaptors/rustgraph.py
@@ -57,17 +57,16 @@ def add_import(
             line_number=line_number,
             line_contents=line_contents,
         )
-        return self._pygraph.add_import(
+        self._pygraph.add_import(
             importer=importer,
             imported=imported,
             line_number=line_number,
             line_contents=line_contents,
         )
 
     def remove_import(self, *, importer: str, imported: str) -> None:
-        self._rustgraph.remove_import(importer=importer, imported=imported)
-        # Can't use rust until have the import details working with rust.
-        return self._pygraph.remove_import(importer=importer, imported=imported)
+        self._pygraph.remove_import(importer=importer, imported=imported)
+        return self._rustgraph.remove_import(importer=importer, imported=imported)
 
     def count_imports(self) -> int:
         return self._rustgraph.count_imports()
@@ -148,7 +147,6 @@ def find_illegal_dependencies_for_layers(
             raise NoSuchContainer(str(e))
 
         return _layers._dependencies_from_tuple(result)
-        # return self._pygraph.find_illegal_dependencies_for_layers(layers, containers)
 
     # Dunder methods
     # --------------