[python] Fix a few bugs introduced with recent changes to make ex_3 work with Talos and meshcat viewer.