get default of false for compile in estimator init

OHDSI · Nov 21, 2024 · ba568ee · ba568ee
1 parent 2b414b8
commit ba568ee
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/inst/python/Estimator.py b/inst/python/Estimator.py
@@ -99,7 +99,8 @@ def __init__(self, model, model_parameters, estimator_settings):
         self.best_score = None
         self.best_epoch = None
         self.learn_rate_schedule = None
-        if estimator_settings["compile"]:
+        torch_compile = estimator_settings.get("compile", False)
+        if torch_compile:
             self.model = torch.compile(self.model, dynamic=False)
 
     def fit(self, dataset, test_dataset):