From 33c6f604ffe822b01596c53c2cdb0f37ccba8088 Mon Sep 17 00:00:00 2001
From: David Muhr <muhrdavid@gmail.com>
Date: Wed, 20 Dec 2023 09:54:13 +0100
Subject: [PATCH] fix usage of MLJModelInterface

---
 Project.toml     | 2 +-
 src/model_def.jl | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/Project.toml b/Project.toml
index eb2edbb..567119a 100644
--- a/Project.toml
+++ b/Project.toml
@@ -1,7 +1,7 @@
 name = "MLJModelInterface"
 uuid = "e80e1ace-859a-464e-9ed9-23947d8ae3ea"
 authors = ["Thibaut Lienart and Anthony Blaom"]
-version = "1.9.3"
+version = "1.9.4"
 
 [deps]
 Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"
diff --git a/src/model_def.jl b/src/model_def.jl
index 17463be..e90d70a 100644
--- a/src/model_def.jl
+++ b/src/model_def.jl
@@ -141,7 +141,7 @@ function _model_constructor(modelname, params, defaults)
         Expr(
             :block,
             Expr(:(=), :model, Expr(:call, :new, params...)),
-            :(message = MLJModelInterface.clean!(model)),
+            :(message = $MLJModelInterface.clean!(model)),
 			:(isempty(message) || @warn message),
 			:(return model)
 		)
@@ -158,7 +158,7 @@ in a model def.
 function _model_cleaner(modelname, defaults, constraints)
     Expr(
         :function,
-        :(MLJModelInterface.clean!(model::$modelname)),
+        :($MLJModelInterface.clean!(model::$modelname)),
         # body of the function
         Expr(
             :block,