From f1f382d9c07e5d01093839775ea0802b9ba3d3d6 Mon Sep 17 00:00:00 2001
From: Edoardo Chidichimo
 <150271881+EdoardoChidichimo@users.noreply.github.com>
Date: Mon, 24 Jun 2024 17:31:01 +0100
Subject: [PATCH] Update hyperit.py

---
 hyperit/hyperit.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/hyperit/hyperit.py b/hyperit/hyperit.py
index ee45985..ca46731 100644
--- a/hyperit/hyperit.py
+++ b/hyperit/hyperit.py
@@ -447,6 +447,9 @@ def __which_estimator(self, measure: str) -> None:
                 self._Calc.setProperty(key, value)
 
         if initialise_parameter:
+            if self._measure == MeasureType.TE:
+                dataDim = self._data1.ndim
+                self._initialise_parameter = (dataDim, dataDim, *initialise_parameter)
             self._initialise_parameter = initialise_parameter
 
         if self._measure == MeasureType.TE and self._estimator_name == 'kernel':