fix assert; update basisname.jl

fixes the assert in predict.jl to make sure partial overlap is not done on models with only one formula. Additionally adds basisname function for uf::UnfoldLinearModel

src/basisfunctions.jl

@@ -220,6 +220,7 @@ kernel(basis::BasisFunction, e) = basis.kernel(e)
 
 basisname(fs::Vector{<:FormulaTerm}) = [name(f.rhs.basisfunction) for f in fs]
 basisname(uf::UnfoldModel) = basisname(formulas(uf))
+basisname(uf::UnfoldLinearModel) = first.(design(uf)) # for linear models we dont save it in the formula
 
 kernel(basis::FIRBasis, e) =
     basis.interpolate ? firkernel(e, basis.times[1:end-1]; interpolate = true) :

src/predict.jl

@@ -201,9 +201,9 @@ function predict(
     kwargs...,
     #eventcolumn = :event,
 )
-    @assert (
-        (!isempty(keep_basis) | !isempty(exclude_basis)) &
-        (basisname(Unfold.formulas(uf))[1] == Any)
+    @assert !(
+        (!isempty(keep_basis) || !isempty(exclude_basis)) &&
+        (length(formulas(uf)) == 1)
     ) "No way to calculate partial overlap if basisnames is Any; please revise model "
 
     @assert !(!isempty(keep_basis) & !isempty(exclude_basis)) "choose either to keep events, or to exclude, but not both"