From 00678c6053900dd53c421ac7e9d935901e864d83 Mon Sep 17 00:00:00 2001
From: William Falcon <falc@fb.com>
Date: Sat, 20 Jul 2019 08:53:36 -0400
Subject: [PATCH] added slurm managed flag catch for non-slurm peeps

---
 pytorch_lightning/models/trainer.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pytorch_lightning/models/trainer.py b/pytorch_lightning/models/trainer.py
index 203a589a..12c501db 100644
--- a/pytorch_lightning/models/trainer.py
+++ b/pytorch_lightning/models/trainer.py
@@ -1,12 +1,12 @@
 """
 The trainer handles all the logic for running a val loop, training loop, distributing, etc...
 """
-from time import sleep
 import subprocess
 import traceback
 import warnings
 import os
 import pdb
+import re
 
 import torch
 from torch.utils.data.distributed import DistributedSampler