tidy imports

pytorchbearer · Nov 5, 2023 · da2c9fe · da2c9fe
1 parent c22ba53
commit da2c9fe
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 2 deletions.
diff --git a/tests/callbacks/test_torch_scheduler.py b/tests/callbacks/test_torch_scheduler.py
@@ -384,7 +384,6 @@ def test_lambda_lr(self, lr_mock):
 
 class TestCyclicLR(TestCase):
     def test_lambda_lr(self):
-        import torch
         if not _pytorch_version_lt("1.0.0"):  # CyclicLR is implemented
             with patch('torch.optim.lr_scheduler.CyclicLR') as lr_mock:
                 state = {torchbearer.OPTIMIZER: 'optimizer', torchbearer.EPOCH: 0, torchbearer.MODEL: Mock()}

diff --git a/torchbearer/callbacks/torch_scheduler.py b/torchbearer/callbacks/torch_scheduler.py
@@ -2,7 +2,6 @@
 import warnings
 
 import torch
-from packaging import version
 
 import torchbearer
 from torchbearer.bases import get_metric, _pytorch_version_lt