Dataclass transform change init False

2025-03-18 00:30:50 +01:00
parent e140523211
commit 999332ef77
3 changed files with 61 additions and 3 deletions
--- a/jedi/plugins/stdlib.py
+++ b/jedi/plugins/stdlib.py
@@ -625,9 +625,17 @@ def _dataclass(value, arguments, callback):
                ]
            )
        elif c.is_function():
-            # dataclass_transform on a decorator equivalent of @dataclass
+            # @dataclass_transform
+            # def create_model(): pass
            return ValueSet([value])
-        elif value.name.string_name != "dataclass_transform":
+        elif (
+            # @dataclass(...)
+            value.name.string_name != "dataclass_transform"
+            # @dataclass_transform
+            # def create_model(): pass
+            # @create_model(...)
+            or isinstance(value, Decoratee)
+        ):
            # dataclass (or like) decorator customization
            return ValueSet(
                [