Update apply_func.py

Author: dbonner

pytorch_lightning/utilities/apply_func.py

@@ -17,7 +17,7 @@
 from copy import copy
 from functools import partial
 from typing import Any, Callable, Optional, Union
-from packaging import version
+from distutils.version import LooseVersion
 
 import numpy as np
 import torch
@@ -27,7 +27,7 @@
 
 if _TORCHTEXT_AVAILABLE:
     from torchtext import __version__ as torchtext_version
-    if version.parse(torchtext_version) < version.parse('0.9.0a0+036df73'):
+    if LooseVersion(torchtext_version) < LooseVersion('0.9.0a0+036df73'):
         from torchtext.data import Batch
     else:
         from torchtext.legacy.data import Batch