Fix: Import of Tensor in bleu unittest.

Labbeti · Mar 4, 2024 · 1a4ba37 · 1a4ba37
1 parent cee41ef
commit 1a4ba37
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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