Merge pull request #19 from JuliaAI/dev

For a 2.1 release

Author: ablaom

Project.toml

@@ -1,7 +1,7 @@
 name = "StatisticalTraits"
 uuid = "64bff920-2084-43da-a3e6-9bb72801c0c9"
 authors = ["Anthony D. Blaom <[email protected]>"]
-version = "2.0.1"
+version = "2.1.0"
 
 [deps]
 ScientificTypesBase = "30f210dd-8aff-4c5f-94ba-8e64358c1161"

README.md

@@ -2,7 +2,7 @@
 
 |  Linux        | Coverage |
 | :-----------: | :------: |
-| [![Build status](https://github.com/alan-turing-institute/StatisticalTraits.jl/workflows/CI/badge.svg)](https://github.com/alan-turing-institute/StatisticalTraits.jl/actions)| [![codecov](https://codecov.io/gh/alan-turing-institute/StatisticalTraits.jl/branch/master/graph/badge.svg?token=rh6HnyIKc1)](https://codecov.io/gh/alan-turing-institute/StatisticalTraits.jl) |
+| [![Build status](https://github.com/JuliaAI/StatisticalTraits.jl/workflows/CI/badge.svg)](https://github.com/JuliaAI/StatisticalTraits.jl/actions)| [![codecov](https://codecov.io/gh/JuliaAI/StatisticalTraits.jl/branch/master/graph/badge.svg?token=rh6HnyIKc1)](https://codecov.io/gh/JuliaAI/StatisticalTraits.jl) |
 
 A light-weight package defining fall-back implementations for a
 collection of traits possessed by statistical objects. Here a "trait"
@@ -19,7 +19,7 @@ prediction_type(::Type) = :unknown # also :probablistic, :deterministic, :interv
 ```
 
 Here `Unknown` is a type defined in the light-weight package
-[ScientificTypes.jl](https://github.com/alan-turing-institute/ScientificTypes.jl),
+[ScientificTypes.jl](https://github.com/JuliaAI/ScientificTypes.jl),
 the only dependency of StatsTraits.jl.
 
 Refer to [source code](src/StatisticalTraits.jl) for a complete list.
@@ -32,7 +32,7 @@ this small package might be of independent interest.
 
 A planned StatisticalMeasures.jl package, to contain measures
 currently in
-[MLJBase.jl](https://github.com/alan-turing-institute/MLJBase.jl),
+[MLJBase.jl](https://github.com/JuliaAI/MLJBase.jl),
 will have this package as a dependency, allowing it to be independent
 of the MLJ packages.
 

src/StatisticalTraits.jl

@@ -9,6 +9,10 @@ const TRAITS = [
     :input_scitype,
     :output_scitype,
     :target_scitype,
+    :fit_data_scitype, #
+    :predict_scitype,  #
+    :transform_scitype, #
+    :inverse_transform_scitype, #
     :is_pure_julia,
     :package_name,
     :package_license,
@@ -24,6 +28,7 @@ const TRAITS = [
     :human_name,
     :is_supervised,
     :prediction_type,
+    :abstract_type, #
     :hyperparameters,
     :hyperparameter_types,
     :hyperparameter_ranges,
@@ -133,6 +138,10 @@ snakecase(s::Symbol) = Symbol(snakecase(string(s)))
 input_scitype(::Type)          = Unknown
 output_scitype(::Type)         = Unknown
 target_scitype(::Type)         = Unknown
+fit_data_scitype(::Type)       = Unknown
+predict_scitype(::Type)        = Unknown
+transform_scitype(::Type)      = Unknown
+inverse_transform_scitype(::Type)  = Unknown
 
 # The following refer to properties of the package defining a type,
 # for use in, say, a registry of machine learning models. All but the
@@ -157,6 +166,7 @@ prediction_type(::Type)        = :unknown
 
 # Miscellaneous:
 
+abstract_type(::Type)   = Any
 is_wrapper(::Type)      = false     # or `true`
 supports_online(::Type) = false     # or `true`
 docstring(M::Type)      = string(M) # some `String`