Fix vector valued functions

JuliaDiff · Aug 26, 2023 · 41c99f4 · 41c99f4
1 parent e19cfe4
commit 41c99f4
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/.devcontainer/devcontainer.json b/.devcontainer/devcontainer.json
@@ -9,7 +9,7 @@
 			]
 		}
 	},
-	"postCreateCommand": "julia -e 'using Pkg; Pkg.add([\"Revise\", \"TestEnv\"])'",
+	"postCreateCommand": "julia -e 'using Pkg; Pkg.add([\"Revise\", \"TestEnv\", \"JuliaFormatter\"])'",
 	"hostRequirements": {
 		"cpus": 4
 	}

diff --git a/src/scalar.jl b/src/scalar.jl
@@ -57,6 +57,10 @@ end
 
 @inline value(t::TaylorScalar) = t.value
 @inline extract_derivative(t::TaylorScalar, i::Integer) = t.value[i]
+@inline function extract_derivative(v::AbstractArray{T},
+    i::Integer) where {T <: TaylorScalar}
+    map(t -> extract_derivative(t, i), v)
+end
 @inline extract_derivative(r, i::Integer) = false
 @inline primal(t::TaylorScalar) = extract_derivative(t, 1)