FileCheck().check("NamedTuple").run(fn2.graph)

class MyMod(torch.nn.Module):

def __init__(self):

super(MyMod, self).__init__()

@torch.jit.unused

def fn(self):

# type: () -> MyTuple

return MyTuple(1)

def forward(self, x):

return self.fn()

mod = torch.jit.script(MyMod())

FileCheck().check_dag("NamedTuple").check_dag("Exception").run(mod.forward.graph)

def test_inherit_method(self):

class A(torch.jit.ScriptModule):

def __init__(self):

if (auto callee =as_function(compiled_fn)) {

return std::make_shared<FunctionValue>(*callee);

}

}

py::bool_ isMethod =py::module::import("inspect").attr("ismethod")(obj);

if (isFunction || isMethod) {

auto rcb =py::module::import("torch.jit").attr("_gen_rcb")(obj,0);

return std::make_shared<PythonValue>(obj, rcb);

}