From 66444aba9609c67113c4aab69d5e79df2a711070 Mon Sep 17 00:00:00 2001
From: Tristan Faine <tfaine@teklia.com>
Date: Wed, 8 Mar 2023 09:08:23 +0100
Subject: [PATCH] fix linting issue
---
dan/predict/attention.py | 5 ++---
1 file changed, 2 insertions(+), 3 deletions(-)
diff --git a/dan/predict/attention.py b/dan/predict/attention.py
index 09229ec0..91ae590e 100644
--- a/dan/predict/attention.py
+++ b/dan/predict/attention.py
@@ -3,7 +3,6 @@ import re
import cv2
import numpy as np
-import math
from PIL import Image
from dan import logger
@@ -74,8 +73,8 @@ def plot_attention(
:param scale: Scaling factor for the output gif image
:param outname: Name of the gif image
"""
-
- height, width, _ = image.shape
+
+ height, width, _ = image.shape
attention_map = []
# Convert to PIL Image and create mask
--
GitLab