[BugFix] Fix CI failures for Python 3.10 and lint

- Use try/except for NotRequired import in test_typedtensordict.py
  (NotRequired moved from typing_extensions to typing in 3.11)
- Add noqa: D103 to dataclass_transform fallback (pydocstyle)
- Remove unused optional_keys variable in _make_init (flake8 F841)

Made-with: Cursor

Author: vmoens

tensordict/typedtensordict.py

@@ -18,7 +18,7 @@
     from typing import dataclass_transform
 except ImportError:
 
-    def dataclass_transform(*args, **kwargs):
+    def dataclass_transform(*args, **kwargs):  # noqa: D103
         def _identity(cls):
             return cls
 
@@ -153,7 +153,6 @@ def _make_init(cls: type) -> callable:
     non_blocking, lock).
     """
     required_keys = cls.__required_keys__
-    optional_keys = cls.__optional_keys__
     expected_keys = cls.__expected_keys__
 
     def __init__(

test/test_typedtensordict.py

@@ -5,7 +5,10 @@
 
 from __future__ import annotations
 
-from typing import NotRequired
+try:
+    from typing import NotRequired
+except ImportError:
+    from typing_extensions import NotRequired
 
 import pytest
 import torch