Masking without breaking graph

Just replying to let you know that you shouldn’t use .data field as it’s only there for backwards compatibility and can lead to incorrect calculations. You can read more here, try using .copy_() instead.