Merge pull request #485 from JuliaTrustworthyAI/475-no-method-matchin…

…g-for-neurotreemodel 475 no method matching for neurotreemodel
JuliaTrustworthyAI · Sep 30, 2024 · 5fc8db7 · 5fc8db7 · pat-alt · Sep 30, 2024
2 parents 2bf113a + 355077b
commit 5fc8db7
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 2 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -6,6 +6,12 @@ The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.1.0/),
 
 *Note*: We try to adhere to these practices as of version [v1.1.1].
 
+## Version [1.3.3] - 2024-09-30
+
+### Changed
+
+- Fixed a remaining bug in `NeuroTreeExt` extensions. [#475]
+
 ## Version [1.3.2] - 2024-09-24
 
 ### Added 

diff --git a/Project.toml b/Project.toml
@@ -1,7 +1,7 @@
 name = "CounterfactualExplanations"
 uuid = "2f13d31b-18db-44c1-bc43-ebaf2cff0be0"
 authors = ["Patrick Altmeyer <p.altmeyer@tudelft.nl> and contributors"]
-version = "1.3.2"
+version = "1.3.3"
 
 [deps]
 CategoricalArrays = "324d7699-5711-5eae-9e2f-1d82baa6b597"

diff --git a/ext/NeuroTreeExt/neurotree.jl b/ext/NeuroTreeExt/neurotree.jl
@@ -83,7 +83,7 @@ logits = Models.logits(M, x) # calculates the logit scores for each output class
 function Models.logits(
     M::Models.Model, type::CounterfactualExplanations.NeuroTreeModel, X::AbstractArray
 )
-    X = X[:, :] |> x -> convert.(eltype(Flux.params(M.fitresult().chain)[1]), x)
+    X = convert(Matrix{Float32}, X[:, :])
     return M.fitresult(X)
 end