fix: PR feedback

src/ssort/_files.py

@@ -2,16 +2,15 @@
 
 import os
 import pathlib
+from functools import cache
 from typing import Iterable
 
 import pathspec
 
-from ssort._utils import memoize
-
 _EMPTY_PATH_SPEC = pathspec.PathSpec([])
 
 
-@memoize
+@cache
 def _is_project_root(path: pathlib.Path) -> bool:
     if path == path.root or path == path.parent:
         return True
@@ -22,7 +21,7 @@ def _is_project_root(path: pathlib.Path) -> bool:
     return False
 
 
-@memoize
+@cache
 def _get_ignore_patterns(path: pathlib.Path) -> pathspec.PathSpec:
     git_ignore = path / ".gitignore"
     if git_ignore.is_file():

src/ssort/_utils.py

@@ -10,8 +10,6 @@
 
 from ssort._exceptions import UnknownEncodingError
 
-memoize = functools.cache
-
 
 def sort_key_from_iter(values):
     index = {statement: index for index, statement in enumerate(values)}

tests/test_ast.py

@@ -12,10 +12,10 @@
     ast.AugStore,
     ast.Param,
     ast.Suite,
+    ast.Index,
+    ast.ExtSlice,
 )
 
-_deprecated_node_types += (ast.Index, ast.ExtSlice)
-
 _ignored_node_types: tuple[type[ast.AST], ...] = (
     ast.expr_context,
     ast.boolop,

tests/test_bindings.py

@@ -6,7 +6,6 @@
 
 from ssort._bindings import get_bindings
 
-
 match_statement = pytest.mark.skipif(
     sys.version_info < (3, 10),
     reason="match statements were introduced in python 3.10",
@@ -935,9 +934,7 @@ def test_if_exp_bindings_walrus_alternate():
 
 
 def test_if_exp_bindings_walrus():
-    node = _parse(
-        "(a := subsequent()) if (b := predicate()) else (c := alternate())"
-    )
+    node = _parse("(a := subsequent()) if (b := predicate()) else (c := alternate())")
     assert list(get_bindings(node)) == ["b", "a", "c"]
 
 
@@ -1069,30 +1066,22 @@ def test_dict_comp_bindings_unpack():
 
 
 def test_dict_comp_bindings_walrus_key():
-    node = _parse(
-        "{(key := item[0]): item[1] for item in iterator if check(item)}"
-    )
+    node = _parse("{(key := item[0]): item[1] for item in iterator if check(item)}")
     assert list(get_bindings(node)) == ["key"]
 
 
 def test_dict_comp_bindings_walrus_value():
-    node = _parse(
-        "{item[0]: (value := item[1]) for item in iterator if check(item)}"
-    )
+    node = _parse("{item[0]: (value := item[1]) for item in iterator if check(item)}")
     assert list(get_bindings(node)) == ["value"]
 
 
 def test_dict_comp_bindings_walrus_iter():
-    node = _parse(
-        "{item[0]: item[1] for item in (it := iterator) if check(item)}"
-    )
+    node = _parse("{item[0]: item[1] for item in (it := iterator) if check(item)}")
     assert list(get_bindings(node)) == ["it"]
 
 
 def test_dict_comp_bindings_walrus_condition():
-    node = _parse(
-        "{item[0]: item[1] for item in iterator if (c := check(item))}"
-    )
+    node = _parse("{item[0]: item[1] for item in iterator if (c := check(item))}")
     assert list(get_bindings(node)) == ["c"]