Merge pull request #194 from JuliaDiff/mz/datatype

Implement `to_vec` for `DataType`s

Author: mzgubic

Project.toml

@@ -1,6 +1,6 @@
 name = "FiniteDifferences"
 uuid = "26cc04aa-876d-5657-8c51-4c34ba976000"
-version = "0.12.18"
+version = "0.12.19"
 
 [deps]
 ChainRulesCore = "d360d2e6-b24c-11e9-a2a3-2a2ae2dbcce4"

src/to_vec.jl

@@ -235,6 +235,13 @@ function to_vec(d::Dict)
     return d_vec, Dict_from_vec
 end
 
+# types
+function FiniteDifferences.to_vec(x::DataType)
+    function DataType_from_vec(x_vec::Vector)
+        return x
+    end
+    return Bool[], DataType_from_vec
+end
 
 # ChainRulesCore Differentials
 function FiniteDifferences.to_vec(x::Tangent{P}) where{P}

test/to_vec.jl

@@ -195,6 +195,11 @@ end
         end
     end
 
+    @testset "DataType" begin
+        test_to_vec(Float64; check_inferred=false) # isa DataType
+        test_to_vec(Vector; check_inferred=false) # isa UnionAll
+    end
+
     @testset "ChainRulesCore Differentials" begin
         @testset "Tangent{Tuple}" begin
             @testset "basic" begin