fix dictionary key check in estimator

OHDSI · Dec 8, 2023 · b85799a · b85799a
1 parent 8fa3711
commit b85799a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/inst/python/Estimator.py b/inst/python/Estimator.py
@@ -19,7 +19,7 @@ def __init__(self, model, model_parameters, estimator_settings):
             self.device = estimator_settings["device"]
 
         torch.manual_seed(seed=self.seed)
-        if estimator_settings["finetune"]:
+        if "finetune" in estimator_settings.keys() and estimator_settings["finetune"]:
             path = estimator_settings["finetune_model_path"]
             fitted_estimator = torch.load(path, map_location="cpu")
             fitted_parameters = fitted_estimator["model_parameters"]

diff --git a/tests/testthat/test-MLP.R b/tests/testthat/test-MLP.R
@@ -123,7 +123,7 @@ test_that("MLP nn-module works ", {
     activation = torch$nn$ReLU,
     normalization = torch$nn$BatchNorm1d,
     dropout = 0.3,
-    d_out = 1L
+    dim_out = 1L
   )
   output <- model(input)
   # model works without numeric variables