From 1a4ba37bf9fc41d85186ed95b9674590dad0b406 Mon Sep 17 00:00:00 2001 From: Labbeti Date: Mon, 4 Mar 2024 18:32:02 +0100 Subject: [PATCH] Fix: Import of Tensor in bleu unittest. --- tests/test_bleu_tchmet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_bleu_tchmet.py b/tests/test_bleu_tchmet.py index d6c8537..13563e6 100644 --- a/tests/test_bleu_tchmet.py +++ b/tests/test_bleu_tchmet.py @@ -2,9 +2,9 @@ # -*- coding: utf-8 -*- import unittest - from unittest import TestCase +from torch import Tensor from torchmetrics.text.bleu import BLEUScore from aac_metrics.classes.bleu import BLEU