search

EEA-sensors / sequential-parallelization-examples

This is a collection of code samples aimed at illustrating temporal parallelization methods for sequential data.
MIT License
28 stars 3 forks source link

small error in kfs() function in parallel-kf-jax #7

Open murphyk opened 2 years ago

murphyk commented 2 years ago

The code currenty defines

def kfs(model, observations):
    return ks(model, *kf(model, observations, False))

but kf only takes 2 args. You should use

def kfs(model, observations):
    return ks(model, *kf(model, observations))