From a4049cb0236155fe088676afaf8e3f4a86adbf68 Mon Sep 17 00:00:00 2001
From: Pherkel
Date: Mon, 18 Sep 2023 21:53:18 +0200
Subject: linter 2

---
 swr2_asr/utils/visualization.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

(limited to 'swr2_asr/utils/visualization.py')

diff --git a/swr2_asr/utils/visualization.py b/swr2_asr/utils/visualization.py
index 23956fd..c56ceb0 100644
--- a/swr2_asr/utils/visualization.py
+++ b/swr2_asr/utils/visualization.py
@@ -15,8 +15,8 @@ def plot(path):
     while True:
         try:
             current_state = torch.load(
-                path + str(epoch), map_location=torch.device("cpu")
-            )  # pylint: disable=no-member
+                path + str(epoch), map_location=torch.device("cpu")  # pylint: disable=no-member
+            )
         except FileNotFoundError:
             break
         train_losses.append((epoch, current_state["train_loss"].item()))
-- 
cgit v1.2.3