Jan

@senysenyseny16 my bad, I forgot the `if __name__ == "__main__"` to actually call the function in the script. I updated the example above. With that, I also get the error...

Thanks! FWIW, this also pertains to (some) other ops where a custom mapping is implemented in this way, for example `Slice`. So it doesn't seem to be specific to the...

Any update on this?