Optimize imports

BloodAxe · BloodAxe · commit 73dccc8157bc · 2020-02-23T15:50:07.000+03:00
diff --git a/pytorch_toolbelt/utils/catalyst/utils.py b/pytorch_toolbelt/utils/catalyst/utils.py
@@ -2,6 +2,8 @@
 
 __all__ = ["clean_checkpoint", "report_checkpoint"]
 
+from catalyst.dl import Callback, CallbackOrder, RunnerState
+
 
 def clean_checkpoint(src_fname, dst_fname):
     """
