Add os import and ensure data directory exists before saving memory 941e057 Kunal Pai commited on May 12