hack to allow picklable streammodel