fix for RuntimeError: Type not yet supported: typing.Literal['sgd', 'adamw', 'adam']

fix for RuntimeError: Type not yet supported: typing.Literal['sgd', 'adamw', 'adam'] #40