Merge pull request #79 from ACEsuit/co/lenbasis

Introduce `basis_size`

src/assemble.jl

@@ -3,6 +3,17 @@ using ParallelDataTransfer
 using ProgressMeter
 using SharedArrays
 
+"""
+    basis_size(model)
+
+Return the length of the basis, assuming that `model` is a linear model, or 
+when interpreted as a linear model. The returned integer must match the 
+size of the feature matrix that will be assembled for the given model.    
+
+It defaults to `Base.length` but can be overloaded if needed.  
+"""
+basis_size(model) = Base.length(model) 
+
 struct DataPacket{T <: AbstractData}
     rows::UnitRange
     data::T
@@ -23,8 +34,8 @@ function assemble(data::AbstractVector{<:AbstractData}, basis)
     packets = DataPacket.(rows, data)
     sort!(packets, by = length, rev = true)
     (nprocs() > 1) && sendto(workers(), basis = basis)
-    @info "  - Creating feature matrix with size ($(rows[end][end]), $(length(basis)))."
-    A = SharedArray(zeros(rows[end][end], length(basis)))
+    @info "  - Creating feature matrix with size ($(rows[end][end]), $(basis_size(basis)))."
+    A = SharedArray(zeros(rows[end][end], basis_size(basis)))
     Y = SharedArray(zeros(size(A, 1)))
     @info "  - Beginning assembly with processor count:  $(nprocs())."
     @showprogress pmap(packets) do p