From d93a64599c29aa84959deb4cd204f73908785b49 Mon Sep 17 00:00:00 2001
From: Stefan Suwelack <stefan.suwelack@renumics.com>
Date: Tue, 19 Sep 2023 13:32:30 +0200
Subject: [PATCH] code linting

---
 renumics/spotlight/layouts/model_compare.py | 4 ++--
 renumics/spotlight/layouts/model_debug.py   | 7 +++++--
 2 files changed, 7 insertions(+), 4 deletions(-)

diff --git a/renumics/spotlight/layouts/model_compare.py b/renumics/spotlight/layouts/model_compare.py
index 75fb5ad1..019b1d94 100644
--- a/renumics/spotlight/layouts/model_compare.py
+++ b/renumics/spotlight/layouts/model_compare.py
@@ -42,7 +42,7 @@ def compare_classification(
     Returns:
         Layout: _description_
     """
-    
+
     # first column: table + issues
     metrics = split(
         [
@@ -118,7 +118,7 @@ def compare_classification(
             )
             column2_list.append(row3)
 
-    column2:Union[Tab, Split]
+    column2: Union[Tab, Split]
 
     if len(column2_list) == 1:
         column2 = column2_list[0]
diff --git a/renumics/spotlight/layouts/model_debug.py b/renumics/spotlight/layouts/model_debug.py
index b5281979..ddb95473 100644
--- a/renumics/spotlight/layouts/model_debug.py
+++ b/renumics/spotlight/layouts/model_debug.py
@@ -1,6 +1,8 @@
 from renumics.spotlight import layout
 from renumics.spotlight.layout import (
     Layout,
+    Tab,
+    Split,
     lenses,
     table,
     similaritymap,
@@ -12,11 +14,10 @@
     confusion_matrix,
     histogram,
 )
-from typing import Optional
+from typing import Optional, Union
 from renumics.spotlight import Audio, Image
 
 
-
 def debug_classification(
     label: str = "label",
     prediction: str = "prediction",
@@ -82,6 +83,8 @@ def debug_classification(
         )
         column2_list.append(row3)
 
+    column2: Union[Tab, Split]
+
     if len(column2_list) == 1:
         column2 = column2_list[0]
     elif len(column2_list) == 2: