fix: ambiguity in target scitype

MilesCranmer · Dec 21, 2024 · a394da2 · a394da2
1 parent 110f1ae
commit a394da2
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/src/MLJInterface.jl b/src/MLJInterface.jl
@@ -622,12 +622,14 @@ MMI.metadata_model(
     human_name="Multi-Target Symbolic Regression via Evolutionary Search",
 )
 
-function MMI.target_scitype(::Type{<:SRRegressor{<:Any,<:Any,<:TemplateExpression}})
+function MMI.target_scitype(
+    ::Type{<:SRRegressor{D,L,E}}
+) where {D<:AbstractDimensions,L,E<:TemplateExpression}
     return AbstractVector{<:Any}
 end
 function MMI.target_scitype(
-    ::Type{<:MultitargetSRRegressor{<:Any,<:Any,<:TemplateExpression}}
-)
+    ::Type{<:MultitargetSRRegressor{D,L,E}}
+) where {D<:AbstractDimensions,L,E<:TemplateExpression}
     return Union{MMI.Table(Any),AbstractMatrix{<:Any}}
 end