pytorch-lightning

5f455902 - Add test for pl_worker_init_function