isort again

Lightning-Universe · Dec 20, 2020 · 63f031c · 63f031c
1 parent ba75ebe
commit 63f031c
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 8 deletions.
diff --git a/pl_bolts/callbacks/verification/__init__.py b/pl_bolts/callbacks/verification/__init__.py
@@ -1,2 +1 @@
-from pl_bolts.callbacks.verification.batch_gradient import BatchGradientVerification
-from pl_bolts.callbacks.verification.batch_gradient import BatchGradientVerificationCallback
+from pl_bolts.callbacks.verification.batch_gradient import BatchGradientVerification, BatchGradientVerificationCallback
diff --git a/pl_bolts/callbacks/verification/batch_gradient.py b/pl_bolts/callbacks/verification/batch_gradient.py
@@ -4,10 +4,7 @@
 from pytorch_lightning.utilities.apply_func import apply_to_collection
 from pytorch_lightning.utilities.exceptions import MisconfigurationException
 
-from pl_bolts.callbacks.verification.base import (
-    VerificationBase,
-    VerificationCallbackBase,
-)
+from pl_bolts.callbacks.verification.base import VerificationBase, VerificationCallbackBase
 
 
 class BatchGradientVerification(VerificationBase):

diff --git a/tests/callbacks/verification/test_batch_gradient.py b/tests/callbacks/verification/test_batch_gradient.py
@@ -1,9 +1,8 @@
 import pytest
 import torch
+from pytorch_lightning import LightningModule, Trainer
 from torch import nn as nn
 
-from pytorch_lightning import Trainer, LightningModule
-
 from pl_bolts.callbacks import BatchGradientVerificationCallback
 from pl_bolts.callbacks.verification import BatchGradientVerification
 from pl_bolts.callbacks.verification.batch_gradient import default_input_mapping, default_output_mapping