From 2b414b8ee30e438f43c5323eb3d1efb29e8d7756 Mon Sep 17 00:00:00 2001
From: egillax <egillax@gmail.com>
Date: Thu, 21 Nov 2024 15:50:32 +0100
Subject: [PATCH] fix for change in branch

---
 inst/python/Estimator.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/inst/python/Estimator.py b/inst/python/Estimator.py
index d9a92ee..69ee153 100644
--- a/inst/python/Estimator.py
+++ b/inst/python/Estimator.py
@@ -99,7 +99,7 @@ def __init__(self, model, model_parameters, estimator_settings):
         self.best_score = None
         self.best_epoch = None
         self.learn_rate_schedule = None
-        if parameters["estimator_settings"]["compile"]:
+        if estimator_settings["compile"]:
             self.model = torch.compile(self.model, dynamic=False)
 
     def fit(self, dataset, test_dataset):