x

awaelchli · Apr 13, 2021 · 4b69a92 · 4b69a92
1 parent 072cd12
commit 4b69a92
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pytorch_lightning/utilities/distributed.py b/pytorch_lightning/utilities/distributed.py
@@ -15,7 +15,7 @@
 import logging
 import os
 import warnings
-from functools import wraps
+from functools import wraps, partial
 from typing import Any, Optional, Union
 
 import torch
@@ -81,6 +81,7 @@ def _debug(*args, **kwargs):
 rank_zero_warn = rank_zero_only(_warn)
 rank_zero_deprecation = partial(rank_zero_warn, category=DeprecationWarning)
 
+
 def find_free_network_port() -> int:
  """
  Finds a free port on localhost.