Skip to content

Commit f868d3a

Browse files
committed
fix the tarragon integration test learner
1 parent 447a519 commit f868d3a

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

test/tarragon.jl

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -111,6 +111,10 @@ LearnAPI.learner(model::TruncatedSVDFitted) = model.learner
111111
LearnAPI.obs(learner::TruncatedSVD, data) = fitobs(learner, data, Tarragon())
112112
LearnAPI.obs(model::TruncatedSVDFitted, data) = obs(model, data, Tarragon())
113113

114+
# training data deconstructor:
115+
LearnAPI.features(learner::TruncatedSVD, data) =
116+
LearnAPI.features(learner, data, Tarragon())
117+
114118
function LearnAPI.fit(learner::TruncatedSVD, observations::Obs; verbosity=1)
115119

116120
x = observations.features # p x n matrix
@@ -198,6 +202,7 @@ model0 = fit(learner, X)
198202
@testset "matrix features for fit" begin
199203
observations = obs(learner, x)
200204
feats = LearnAPI.features(learner, observations)
205+
@test LearnAPI.features(learner, x).features == feats.features
201206
@test feats isa Obs
202207
@test feats.features == x
203208
@test feats.names == [:x1, :x2, :x3, :x4, :x5]

0 commit comments

Comments
 (0)