90ed76e
1
2
3
4
5
6
7
8
9
10
11
12
13
14
import pickle load = pickle.load class Empty: pass class Unpickler(pickle.Unpickler): def find_class(self, module, name): #TODO: safe unpickle if module.startswith("pytorch_lightning"): return Empty return super().find_class(module, name)