Você quase conseguiu:
import json
def remove_dot_key(obj):
for key in obj.keys():
new_key = key.replace(".","")
if new_key != key:
obj[new_key] = obj[key]
del obj[key]
return obj
new_json = json.loads(data, object_hook=remove_dot_key)
Você estava retornando um dicionário dentro do seu loop, então só modificaria uma chave. E você não precisa fazer uma cópia dos valores, apenas renomeie as chaves.