From 055c1ba964536ca97e13a5858d9df58f3940f808 Mon Sep 17 00:00:00 2001
From: Francesco Andreuzzi <andreuzzi.francesco@gmail.com>
Date: Thu, 8 Dec 2022 22:13:08 +0100
Subject: [PATCH] fix

---
 pina/utils.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/pina/utils.py b/pina/utils.py
index ed18af0..429985a 100644
--- a/pina/utils.py
+++ b/pina/utils.py
@@ -1,6 +1,8 @@
 """Utils module"""
 from functools import reduce
 
+from .label_tensor import LabelTensor
+
 
 def number_parameters(model, aggregate=True, only_trainable=True): #TODO: check
     """