[notebooks] Remove TODO