Files
zenml/venv/lib/python3.9/site-packages/asttokens/astroid_compat.py
Christian Mantha 2ca0b9ef7c star
2026-03-02 19:10:52 -05:00

19 lines
586 B
Python

try:
from astroid import nodes as astroid_node_classes
# astroid_node_classes should be whichever module has the NodeNG class
from astroid.nodes import NodeNG
from astroid.nodes import BaseContainer
except Exception:
try:
from astroid import node_classes as astroid_node_classes
from astroid.node_classes import NodeNG
from astroid.node_classes import _BaseContainer as BaseContainer
except Exception: # pragma: no cover
astroid_node_classes = None
NodeNG = None
BaseContainer = None
__all__ = ["astroid_node_classes", "NodeNG", "BaseContainer"]