Still debugging the theanoized code.