Unfortunately, no. The resin content of the products is high enough to effectively have them function somewhat as sealers, and the stain will not penetrate like it will into the bare wood.
The suggested workaround is to mix dry pigments or house paint colorants into the paste or putty as you combine the Part A and B. This can approximate the color of the stained wood around it.