fixed tests

Lightning-AI · Nov 29, 2019 · db0587f · db0587f
1 parent 6629897
commit db0587f
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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