Fix loading astroid.nodes.node_classes for astroid>3.0

gwax · May 10, 2024 · e5b1a37 · e5b1a37
1 parent 9598a8a
commit e5b1a37
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/pylint_sqlalchemy/dml_no_argument.py b/pylint_sqlalchemy/dml_no_argument.py
@@ -1,6 +1,9 @@
 """Linter to fix missing `dml` argument for pylint and sqlalchemy."""
 import astroid
-import astroid.node_classes
+try:
+    import astroid.node_classes
+except ModuleNotFoundError:
+    import astroid.nodes.node_classes
 
 
 def strip_dml(node):