From f8ef3b04a58b85c5874553ad9b0647ba498018f8 Mon Sep 17 00:00:00 2001
From: Prabhu Teja <prbuteja@amazon.de>
Date: Thu, 3 Aug 2023 16:29:08 +0200
Subject: [PATCH] flake8

---
 src/renate/updaters/avalanche/model_updater.py  | 2 +-
 src/renate/updaters/experimental/er.py          | 2 +-
 src/renate/updaters/experimental/fine_tuning.py | 2 +-
 src/renate/updaters/experimental/gdumb.py       | 2 +-
 src/renate/updaters/experimental/joint.py       | 2 +-
 src/renate/updaters/experimental/offline_er.py  | 2 +-
 src/renate/updaters/model_updater.py            | 2 +-
 7 files changed, 7 insertions(+), 7 deletions(-)

diff --git a/src/renate/updaters/avalanche/model_updater.py b/src/renate/updaters/avalanche/model_updater.py
index 3723661c..68438f61 100644
--- a/src/renate/updaters/avalanche/model_updater.py
+++ b/src/renate/updaters/avalanche/model_updater.py
@@ -2,7 +2,7 @@
 # SPDX-License-Identifier: Apache-2.0
 import logging
 from pathlib import Path
-from typing import Any, Callable, Dict, List, Optional, Type, Union
+from typing import Any, Callable, Dict, List, Optional, Type
 
 import torch
 import torchmetrics
diff --git a/src/renate/updaters/experimental/er.py b/src/renate/updaters/experimental/er.py
index 19878297..f9e6da71 100644
--- a/src/renate/updaters/experimental/er.py
+++ b/src/renate/updaters/experimental/er.py
@@ -2,7 +2,7 @@
 # SPDX-License-Identifier: Apache-2.0
 import abc
 from functools import partial
-from typing import Any, Callable, Dict, List, Optional, Tuple, Union
+from typing import Any, Callable, Dict, List, Optional, Tuple
 
 import torch
 import torchmetrics
diff --git a/src/renate/updaters/experimental/fine_tuning.py b/src/renate/updaters/experimental/fine_tuning.py
index b7024b54..f31295dd 100644
--- a/src/renate/updaters/experimental/fine_tuning.py
+++ b/src/renate/updaters/experimental/fine_tuning.py
@@ -1,7 +1,7 @@
 # Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
 # SPDX-License-Identifier: Apache-2.0
 from functools import partial
-from typing import Callable, Dict, List, Optional, Union
+from typing import Callable, Dict, List, Optional
 
 import torch
 import torchmetrics
diff --git a/src/renate/updaters/experimental/gdumb.py b/src/renate/updaters/experimental/gdumb.py
index 8f7d0da5..d0d69656 100644
--- a/src/renate/updaters/experimental/gdumb.py
+++ b/src/renate/updaters/experimental/gdumb.py
@@ -1,7 +1,7 @@
 # Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
 # SPDX-License-Identifier: Apache-2.0
 from functools import partial
-from typing import Any, Callable, Dict, List, Optional, Tuple, Union
+from typing import Any, Callable, Dict, List, Optional, Tuple
 
 import torch
 import torchmetrics
diff --git a/src/renate/updaters/experimental/joint.py b/src/renate/updaters/experimental/joint.py
index 22bc9c7b..cf907560 100644
--- a/src/renate/updaters/experimental/joint.py
+++ b/src/renate/updaters/experimental/joint.py
@@ -2,7 +2,7 @@
 # SPDX-License-Identifier: Apache-2.0
 import os
 from functools import partial
-from typing import Any, Callable, Dict, List, Optional, Tuple, Union
+from typing import Any, Callable, Dict, List, Optional, Tuple
 
 import torch
 import torchmetrics
diff --git a/src/renate/updaters/experimental/offline_er.py b/src/renate/updaters/experimental/offline_er.py
index 7b7a9e27..17c3e0e1 100644
--- a/src/renate/updaters/experimental/offline_er.py
+++ b/src/renate/updaters/experimental/offline_er.py
@@ -1,7 +1,7 @@
 # Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
 # SPDX-License-Identifier: Apache-2.0
 from functools import partial
-from typing import Any, Callable, Dict, List, Optional, Tuple, Union
+from typing import Any, Callable, Dict, List, Optional, Tuple
 
 import torch
 import torchmetrics
diff --git a/src/renate/updaters/model_updater.py b/src/renate/updaters/model_updater.py
index 93736d35..c340abde 100644
--- a/src/renate/updaters/model_updater.py
+++ b/src/renate/updaters/model_updater.py
@@ -4,7 +4,7 @@
 import logging
 import warnings
 from pathlib import Path
-from typing import Any, Callable, Dict, List, Optional, Type, Union
+from typing import Any, Callable, Dict, List, Optional, Type
 
 import torch
 import torchmetrics