fixed black

chintu619 · May 11, 2022 · ec7e2b0 · ec7e2b0
1 parent 2be9ddc
commit ec7e2b0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/espnet2/gan_tts/vits/vits.py b/espnet2/gan_tts/vits/vits.py
@@ -42,9 +42,9 @@
 if LooseVersion(torch.__version__) >= LooseVersion("1.6.0"):
     from torch.cuda.amp import autocast
 else:
+    # Nothing to do if torch<1.6.0
     @contextmanager
-    def autocast(enabled=True):
-        """Define dummy function to be compatible with torch < 1.6.0."""
+    def autocast(enabled=True):  # NOQA
         yield