adjust setter to include type annotation

This commit is contained in:
Khaleel Al-Adhami 2025-01-31 19:19:21 -08:00
parent 238b03a8c7
commit 1f5c2b67b6

View File

@ -933,7 +933,7 @@ class Var(Generic[VAR_TYPE]):
"""
actual_name = self._var_field_name
def setter(state: BaseState, value: Any):
def setter(state: Any, value: Any):
"""Get the setter for the var.
Args:
@ -951,6 +951,8 @@ class Var(Generic[VAR_TYPE]):
else:
setattr(state, actual_name, value)
setter.__annotations__["value"] = self._var_type
setter.__qualname__ = self._get_setter_name()
return setter