Update tests/checkpointing/test_model_checkpoint.py

Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>
Lightning-AI · Nov 20, 2020 · 5851762 · 5851762
1 parent bb802e2
commit 5851762
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/tests/checkpointing/test_model_checkpoint.py b/tests/checkpointing/test_model_checkpoint.py
@@ -27,8 +27,7 @@
 import cloudpickle
 import pytest
 import torch
-from omegaconf import Container
-from omegaconf import OmegaConf
+from omegaconf import Container, OmegaConf
 from argparse import Namespace
 
 import tests.base.develop_utils as tutils