From d5373154fdd8de7c83daf7888228183785a0c23c Mon Sep 17 00:00:00 2001
From: manonBlanco <blanco@teklia.com>
Date: Mon, 12 Jun 2023 16:02:40 +0200
Subject: [PATCH] Replace utils pairwise function by itertools pariwise
---
dan/predict/prediction.py | 3 ++-
dan/utils.py | 12 ------------
2 files changed, 2 insertions(+), 13 deletions(-)
diff --git a/dan/predict/prediction.py b/dan/predict/prediction.py
index ce25c133..b1e8a356 100644
--- a/dan/predict/prediction.py
+++ b/dan/predict/prediction.py
@@ -2,6 +2,7 @@
import os
import pickle
+from itertools import pairwise
from pathlib import Path
import cv2
@@ -20,7 +21,7 @@ from dan.predict.attention import (
plot_attention,
split_text_and_confidences,
)
-from dan.utils import pairwise, read_image
+from dan.utils import read_image
class DAN:
diff --git a/dan/utils.py b/dan/utils.py
index 7325b2b3..93243fcf 100644
--- a/dan/utils.py
+++ b/dan/utils.py
@@ -1,6 +1,4 @@
# -*- coding: utf-8 -*-
-from itertools import tee
-
import cv2
import numpy as np
import torch
@@ -154,13 +152,3 @@ def round_floats(float_list, decimals=2):
Round list of floats with fixed decimals
"""
return [np.around(num, decimals) for num in float_list]
-
-
-def pairwise(iterable):
- """
- Not necessary when using 3.10. See https://docs.python.org/3/library/itertools.html#itertools.pairwise.
- """
- # pairwise('ABCDEFG') --> AB BC CD DE EF FG
- a, b = tee(iterable)
- next(b, None)
- return zip(a, b)
--
GitLab