Yes it works fine for me. Make sure your inputs have the same name as the custom code and that you’re passing different values into each input. One input will output to the rastered image and the other will output to the lumen scene.
I demonstrated it working on this post: