




"""simple docstring"""




import pickle
import unittest

import torch

from accelerate import Accelerator
from accelerate.state import AcceleratorState
from accelerate.test_utils import require_cpu






@require_cpu
class     __magic_name__							(       unittest.TestCase				):






				'''simple docstring'''





				def    _lowerCAmelCase				(       self ):





							"""simple docstring"""
							lowerCamelCase			      =					torch.nn.Linear(10		, 10 )
							lowerCamelCase			      =					torch.optim.SGD(model.parameters()		, 0.1 )
							lowerCamelCase			      =					Accelerator()
							lowerCamelCase			      =					accelerator.prepare(lowerCAmelCase_ )
							try:
										pickle.loads(pickle.dumps(lowerCAmelCase_ ) )
							except Exception as e:
										self.fail(f'Accelerated optimizer pickling failed with {e}' )
							AcceleratorState._reset_state()





'''simple docstring'''







def 				SCREAMING_SNAKE_CASE__	(   __A = 100	)       ->	int:
	_snake_case   						=		n * (n + 1) * (2 * n + 1) / 6
	_snake_case   						=		(n * (n + 1) / 2) ** 2
	return int(square_of_sum - sum_of_squares	)


if __name__ == "__main__":
					print(F'''{solution() = }''')

