[BugFix] Fix version parsing in extensions

ghstack-source-id: 903f2b01b508b81b1b4f92c4297d390da79fe8a2 Pull Request resolved: #2542
pytorch · Nov 6, 2024 · 997d90e · 997d90e
1 parent 0a13cbd
commit 997d90e
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/torchrl/_extension.py b/torchrl/_extension.py
@@ -7,7 +7,11 @@
 import warnings
 
 from packaging.version import parse
-from torchrl import __version__
+
+try:
+    from .version import __version__
+except ImportError:
+    __version__ = None
 
 
 def is_module_available(*modules: str) -> bool:
@@ -28,6 +32,8 @@ def _init_extension():
 
 
 def _is_nightly(version):
+    if version is None:
+        return True
     parsed_version = parse(version)
     return parsed_version.local is not None