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 ba5deb4c162ca..67821ae95bb30 100644 --- a/pytorch_lightning/trainer/ddp_mixin.py +++ b/pytorch_lightning/trainer/ddp_mixin.py @@ -116,6 +116,7 @@ def train_fx(trial_hparams, cluster_manager, _): import os import re import logging +import warnings import torch