(self, *args, **kwargs)
| 156 | def remove_vgg(fn): |
| 157 | @wraps(fn) |
| 158 | def inner(self, *args, **kwargs): |
| 159 | has_vgg = hasattr(self, "vgg") |
| 160 | if has_vgg: |
| 161 | vgg = self.vgg |
| 162 | delattr(self, "vgg") |
| 163 | |
| 164 | out = fn(self, *args, **kwargs) |
| 165 | |
| 166 | if has_vgg: |
| 167 | self.vgg = vgg |
| 168 | |
| 169 | return out |
| 170 | |
| 171 | return inner |
| 172 |