Solving Magic Square

from random import shuffle

def colsCheck(MS):
    if (MS[0][0]+MS[1][0]+MS[2][0] == 15) and (MS[0][1]+MS[1][1]+MS[2][1] == 15) and (MS[0][2]+MS[1][2]+MS[2][2] == 15):
        return True
    else:
        return False

def diagCheck(MS):
    if (MS[0][0]+MS[1][1]+MS[2][2] == 15) and (MS[0][2]+MS[1][1]+MS[2][0] == 15):
        return True
    else:
        return False

def rowsCheck(MS):
    if (sum(MS[0]) == 15) and (sum(MS[1]) == 15) and (sum(MS[2]) == 15):
        return True
    else:
        return False

nos = [i for i in range(1, 10)]
MS = [[0, 0, 0], [0, 0, 0], [0, 0, 0]]

while True:
    for i in MS:
        for j in range(len(i)):
            shuffle(nos)
            i[j] = nos.pop()

    if colsCheck(MS) and rowsCheck(MS) and diagCheck(MS):
        for i in MS:
            for j in i:
                print(f"{j} ", end="")
            print()

        break
    else:
        nos = [i for i in range(1, 10)]
        MS = [[0, 0, 0], [0, 0, 0], [0, 0, 0]]