cleanup

drivendataorg · May 9, 2023 · 754afa2 · 754afa2
1 parent c3e08e2
commit 754afa2
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 3 deletions.
diff --git a/tests/test_instantiate_model.py b/tests/test_instantiate_model.py
@@ -103,8 +103,6 @@ def test_not_use_default_model_labels(dummy_trained_model_checkpoint):
         dummy_trained_model_checkpoint
     )
 
-    # instantiate_model returns model on GPU if checkpoint is from PTL >= 2.0, was
-    # trained on GPU, and default model labels are being used (not the case here)
     model = instantiate_model(
         checkpoint=dummy_trained_model_checkpoint,
         scheduler_config="default",

diff --git a/zamba/models/model_manager.py b/zamba/models/model_manager.py
@@ -14,7 +14,6 @@
 from pytorch_lightning.loggers import TensorBoardLogger
 from pytorch_lightning.strategies import DDPStrategy
 from pytorch_lightning.tuner import Tuner
-import torch
 
 from zamba.data.video import VideoLoaderConfig
 from zamba.models.config import (