| 137 | |
| 138 | |
| 139 | class WrapperLeafModule(nn.Module): |
| 140 | |
| 141 | def __init__(self, hidden_dim): |
| 142 | super(WrapperLeafModule, self).__init__() |
| 143 | self.child = SubLeafModule(hidden_dim) |
| 144 | |
| 145 | def forward(self, x): |
| 146 | return self.child(x) |
| 147 | |
| 148 | |
| 149 | def test_set_leaf_modules_with_fully_qualified_name(): |
no outgoing calls