o zi@sTddlZddlmZddlZejdgdddZejdddgd d ZdS) N) Optimizeroptimizer_name)yogiYogiYoGicCs2t|}t|ts J|j|ksJdSN)optimget issubclassr__name__lower)r optimizer_clsrD/home/ubuntu/.local/lib/python3.10/site-packages/tests/test_utils.pytest_returns_optimizer_clss rfoobarcCs:ttt|WddS1swYdSr)pytestraises ValueErrorrr )rrrr test_raisess  "r) rtorch.optim.optimizerrtorch_optimizerrmark parametrizerrrrrrs