def _make_lists()

in grok/data.py [0:0]

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

    def _make_lists(cls, sizes=[2, 3], nums=NUMS):
        lists: dict = {}
        for size in sizes:
            lists[size] = torch.tensor(
                list(itertools.permutations(nums, r=size)),
                dtype=torch.int,
            )
        return lists