Update

[ghstack-poisoned]
pytorch · Nov 18, 2024 · 8df6918 · 8df6918
2 parents 0ef9a38 + d8fea5c
commit 8df6918
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/torchrl/modules/distributions/continuous.py b/torchrl/modules/distributions/continuous.py
@@ -42,9 +42,8 @@
 except ImportError:
     from torch._dynamo import is_compiling as is_dynamo_compiling
 
-TORCH_VERSION_PRE_2_6 = version.parse(torch.__version__).base_version < version.parse(
-    "2.6.0"
-)
+TORCH_VERSION = version.parse(torch.__version__).base_version
+TORCH_VERSION_PRE_2_6 = version.parse(TORCH_VERSION) < version.parse("2.6.0")
 
 
 class IndependentNormal(D.Independent):