wip

Author: jacobtylerwalls

astroid/brain/brain_collections.py

@@ -20,8 +20,7 @@
 
 def _collections_transform():
     return parse(
-        ("    import _collections_abc as abc" if PY313_PLUS and not PY313_0 else "")
-        + """
+        """
     class defaultdict(dict):
         default_factory = None
         def __missing__(self, key): pass
@@ -40,6 +39,11 @@ def _collections_abc_313_0_transform() -> nodes.Module:
     )
 
 
+def _collections_abc_313_1_transform() -> nodes.Module:
+    """See https://github.com/python/cpython/pull/124735"""
+    return AstroidBuilder(AstroidManager()).string_build("from collections import abc")
+
+
 def _deque_mock():
     base_deque_class = """
     class deque(object):
@@ -137,3 +141,7 @@ def register(manager: AstroidManager) -> None:
         register_module_extender(
             manager, "collections.abc", _collections_abc_313_0_transform
         )
+    elif PY313_PLUS:
+        register_module_extender(
+            manager, "_collections_abc", _collections_abc_313_1_transform
+        )