From 3e0ba77315642d5ba73aca0bed3321952992816b Mon Sep 17 00:00:00 2001
From: Vincent Moens <vmoens@meta.com>
Date: Wed, 6 Nov 2024 10:51:29 +0000
Subject: [PATCH] Update

[ghstack-poisoned]
---
 torchrl/_extension.py | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/torchrl/_extension.py b/torchrl/_extension.py
index 6079e63d4ed..61eedb46418 100644
--- 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