def __setstate__()

in grok/training.py [0:0]

• 4 lines of code
• 2 McCabe index (conditional complexity)

    def __setstate__(self, state):
        super(CustomAdamW, self).__setstate__(state)
        for group in self.param_groups:
            group.setdefault("amsgrad", False)