Use learn on one cell temporal memory

cc: @subutai If you pull the latest (after this PR is merged), you can remove the hack to `temporal_memory.py` and the tests will still work.