以下の辞書内包表記がエラーになります。
どうすればいいでしょうか?

import math
def normalize_coefficients(n, D):
return {(s, i):0.5*math.sqrt(n/s) if s != 0 else (s, i):math.sqrt(n) for (s, i) in D.keys()}