Ojas Modi
Ojas Modi
> Adding the `output_key` as above worked for me also. Actually it would work for every type of memory object.