Try adding
if len(bids_.shape) < 2:
            bids_ = bids_.unsqueeze(dim=0)

between lines 50 and 51 in auction_lap.py:
bids_.zero_()
if len(bids_.shape) < 2:
            bids_ = bids_.unsqueeze(dim=0)
bids_.scatter_(
            dim=1,
            index=first_idx.contiguous().view(-1, 1),
            src=bid_increments.view(-1, 1)
        )

edit:
You will also want make the following adjustment at line 40:
unassigned = (curr_ass == -1).nonzero().squeeze()
if len(unassigned.shape) < 1:
            unassigned = unassigned.unsqueeze(0)
value = X[unassigned] - cost