tweak

JuliaAI · Oct 8, 2024 · f690c44 · f690c44
1 parent 5afe7ac
commit f690c44
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/traits.jl b/test/traits.jl
@@ -6,7 +6,7 @@ using LearnAPI
 # does nothing useful
 struct SmallAlgorithm end
 LearnAPI.fit(algorithm::SmallAlgorithm, data; verbosity=1) = algorithm
-LearnAPI.algorithm(algorithm::SmallAlgorithm) = algorithm
+LearnAPI.algorithm(model::SmallAlgorithm) = model
 @trait(
     SmallAlgorithm,
     constructor = SmallAlgorithm,