limit signature

Author: adamchainz

src/django_mysql/models/fields/dynamic.py

@@ -323,12 +323,8 @@ def __init__(
         data_type: str,
         *expressions: Any,
         subspec: SpecDict | None = None,
-        output_field: Field[Any, Any] | None = None,
-        **extra: Any,
     ) -> None:
-        if output_field is not None:
-            raise ValueError("Cannot set output_field for KeyTransform")
-
+        output_field: Field[Any, Any]
         if data_type == "BINARY":
             output_field = DynamicField(spec=subspec)
         else:
@@ -337,7 +333,7 @@ def __init__(
             except KeyError:
                 raise ValueError(f"Invalid data_type {data_type!r}")
 
-        super().__init__(*expressions, output_field=output_field, **extra)
+        super().__init__(*expressions, output_field=output_field)
 
         self.key_name = key_name
         self.data_type = data_type

tests/testapp/test_dynamicfield.py

@@ -160,12 +160,6 @@ def test_non_existent_transform(self):
     def test_has_key(self):
         assert list(DynamicModel.objects.filter(attrs__has_key="c")) == self.objs[1:3]
 
-    def test_key_transform_initialize_output_field(self):
-        with pytest.raises(ValueError) as excinfo:
-            KeyTransform("x", "y", output_field=CharField())
-
-        assert str(excinfo.value) == "Cannot set output_field for KeyTransform"
-
     def test_key_transform_initialize_bad_type(self):
         with pytest.raises(ValueError) as excinfo:
             KeyTransform("x", "unknown")