diff --git a/torch/csrc/jit/python/python_ir.cpp b/torch/csrc/jit/python/python_ir.cpp index 3d11934c053..69db1ff2bf4 100644 --- a/torch/csrc/jit/python/python_ir.cpp +++ b/torch/csrc/jit/python/python_ir.cpp @@ -1071,8 +1071,15 @@ void initPythonIRBindings(PyObject* module_) { return get_python_cu()->get_class(c10::QualifiedName(qualified_name)); })) .def("name", [](ClassType& self) { return self.name()->name(); }) - .def("qualified_name", [](ClassType& self) { - return self.name()->qualifiedName(); + .def( + "qualified_name", + [](ClassType& self) { return self.name()->qualifiedName(); }) + .def("method_names", [](ClassType& self) { + std::vector method_names; + for (const auto* method : self.methods()) { + method_names.push_back(method->name()); + } + return method_names; }); py::class_(m, "EnumType") .def(py::init([](const std::string& qualified_name,