Fix import

Author: SeanNaren

tests/deprecated_api/test_remove_1-5.py

@@ -19,6 +19,7 @@
 
 import pytest
 import torch
+from torch import optim
 
 from pytorch_lightning import Callback, Trainer
 from pytorch_lightning.callbacks import ModelCheckpoint