Add hydra experimental to correct location

Author: SeanNaren

pytorch_lightning/utilities/__init__.py

@@ -31,6 +31,7 @@
     _GROUP_AVAILABLE,
     _HOROVOD_AVAILABLE,
     _HYDRA_AVAILABLE,
+    _HYDRA_EXPERIMENTAL_AVAILABLE,
     _module_available,
     _NATIVE_AMP_AVAILABLE,
     _OMEGACONF_AVAILABLE,

pytorch_lightning/utilities/imports.py

@@ -45,6 +45,7 @@ def _module_available(module_path: str) -> bool:
 _NATIVE_AMP_AVAILABLE = _module_available("torch.cuda.amp") and hasattr(torch.cuda.amp, "autocast")
 _OMEGACONF_AVAILABLE = _module_available("omegaconf")
 _HYDRA_AVAILABLE = _module_available("hydra")
+_HYDRA_EXPERIMENTAL_AVAILABLE = _module_available("hydra.experimental")
 _HOROVOD_AVAILABLE = _module_available("horovod.torch")
 _TORCHTEXT_AVAILABLE = _module_available("torchtext")
 _XLA_AVAILABLE = _module_available("torch_xla")

tests/models/test_hparams.py

@@ -33,6 +33,7 @@
 if _HYDRA_EXPERIMENTAL_AVAILABLE:
     from hydra.experimental import compose, initialize
 
+
 class SaveHparamsModel(BoringModel):
     """ Tests that a model can take an object """
     def __init__(self, hparams):