formatting

pytorch · Jul 7, 2023 · bb4608c · bb4608c
1 parent 28a97a1
commit bb4608c
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/references/classification/presets.py b/references/classification/presets.py
@@ -6,9 +6,11 @@ def get_module(use_v2):
     # We need a protected import to avoid the V2 warning in case just V1 is used
     if use_v2:
         import torchvision.transforms.v2
+
         return torchvision.transforms.v2
     else:
         import torchvision.transforms
+
         return torchvision.transforms