File tree Expand file tree Collapse file tree 2 files changed +9
-3
lines changed Expand file tree Collapse file tree 2 files changed +9
-3
lines changed Original file line number Diff line number Diff line change @@ -36,8 +36,10 @@ docstring(M::Type{<:MLJType}) = name(M)
36
36
docstring (M:: Type{<:Model} ) = " $(name (M)) from $(package_name (M)) .jl.\n " *
37
37
" [Documentation]($(package_url (M)) )."
38
38
# "derived" traits:
39
- name (M:: Type ) = string (M)
40
- name (M:: Type{<:MLJType} ) = split (string (coretype (M)), ' .' )[end ] |> String
39
+ typename (s) = replace (s, r" typename\( (.*?)\) " => s "\1 " )
40
+ name (M:: Type ) = string (M) |> typename
41
+ name (M:: Type{<:MLJType} ) = split (string (coretype (M)), ' .' )[end ] |> String |> typename
42
+
41
43
42
44
is_supervised (:: Type ) = false
43
45
is_supervised (:: Type{<:Supervised} ) = true
Original file line number Diff line number Diff line change @@ -72,5 +72,9 @@ struct FooMeasure <: MLJType end
72
72
@test name (Float64) == " Float64"
73
73
74
74
df = DataFrame (a= randn (2 ), b= randn (2 ))
75
- @test string (M. coretype (typeof (df))) == " DataFrame"
75
+ if VERSION < v " 1.6-"
76
+ @test string (M. coretype (typeof (df))) == " DataFrame"
77
+ else
78
+ @test string (M. coretype (typeof (df))) == " typename(DataFrame)"
79
+ end
76
80
end
You can’t perform that action at this time.
0 commit comments