FlatMap no longer shows User function name

def my_function(element):
    ...
pcoll | beam.FlatMap(my_function)

now has a stage name of FlatMap(<lambda at core.py:2027>) rather than FlatMap(my_function). Not sure when this regression was introduced.