I think this part can be dropped and the class can just be referenced directly in the Python code? We do `load_class_by_name` in places were we need to be able to plugin a new class at run time but I don't think this is one of those cases?