Fixed import for importlib

torchrl/_extension.py

@@ -3,18 +3,17 @@
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
 
-import importlib
+import importlib.util
 import warnings
 
 
 def is_module_available(*modules: str) -> bool:
-    r"""Returns if a top-level module with :attr:`name` exists *without** importing it.
+    """Returns if a top-level module with :attr:`name` exists *without** importing it.
 
     This is generally safer than try-catch block around a
     `import X`. It avoids third party libraries breaking assumptions of some of
     our tests, e.g., setting multiprocessing start method when imported
     (see librosa/#747, torchvision/#544).
-
     """
     return all(importlib.util.find_spec(m) is not None for m in modules)