Switched libcst._nodes.statement imports to cst. imports

Instagram · Jun 5, 2024 · 7368528 · 7368528
1 parent fb27778
commit 7368528
Showing 1 changed file with 2 additions and 7 deletions.
diff --git a/src/fixit/rules/deprecated_abc_import.py b/src/fixit/rules/deprecated_abc_import.py
@@ -8,11 +8,6 @@
 import libcst as cst
 
 import libcst.matchers as m
-from libcst._nodes.statement import (
-    BaseCompoundStatement,
-    ImportFrom,
-    SimpleStatementLine,
-)
 
 from fixit import Invalid, LintRule, Valid
 
@@ -147,8 +142,8 @@ def visit_ImportFrom(self, node: cst.ImportFrom) -> None:
                 )
 
     def get_import_from(
-        self, node: Union[SimpleStatementLine, BaseCompoundStatement]
-    ) -> Optional[ImportFrom]:
+        self, node: Union[cst.SimpleStatementLine, cst.BaseCompoundStatement]
+    ) -> Optional[cst.ImportFrom]:
         """
         Iterate over a Statement Sequence and return a Statement if it is a
         `cst.ImportFrom` statement.