Clean up pylint warnings in TemporalMemory