textgrad.optimizer.optimizer_prompts.py lacks one argument for class TextualGradientDescentwithMomentum(Optimizer) #157
Description
In class TextualGradientDescentwithMomentum(Optimizer): for def _update_prompt(self, variable: Variable, momentum_storage_idx: int):
the function does not return anything. It needs to have "return prompt" otherwise in step(self), prompt_update_parameter = self._update_prompt(parameter, momentum_storage_idx=idx), prompt_update_parameter would be null.