From db0587f15823a7ef49f953642abf35f3d73e8324 Mon Sep 17 00:00:00 2001 From: williamFalcon Date: Thu, 28 Nov 2019 16:02:36 -0800 Subject: [PATCH] fixed tests --- pytorch_lightning/trainer/ddp_mixin.py | 1 + 1 file changed, 1 insertion(+) diff --git a/pytorch_lightning/trainer/ddp_mixin.py b/pytorch_lightning/trainer/ddp_mixin.py index ba5deb4c..67821ae9 100644 --- a/pytorch_lightning/trainer/ddp_mixin.py +++ b/pytorch_lightning/trainer/ddp_mixin.py @@ -116,6 +116,7 @@ When the script starts again, Lightning will: import os import re import logging +import warnings import torch