
"""simple docstring"""


def    _snake_case					(		snake_case__ :  int  ):
	assert (
	    isinstance(snake_case__						,   snake_case__  ) and number_of_steps > 0
	), F'number_of_steps needs to be positive integer, your input {number_of_steps}'
	if number_of_steps == 1:
		return 1
	A ,   A     = 1, 1
	for _ in range(number_of_steps - 1  ):
		A ,   A     = current + previous, current
	return current


if __name__ == "__main__":
		import doctest

		doctest.testmod()

"""simple docstring"""


from torch import nn






def    _snake_case					(		snake_case__ :  Union[str, Any]  ):
	if act_fn in ["swish", "silu"]:
		return nn.SiLU()
	elif act_fn == "mish":
		return nn.Mish()
	elif act_fn == "gelu":
		return nn.GELU()
	else:
		raise ValueError(F'Unsupported activation function: {act_fn}'  )