[BugFix] Fix nightly tensordict dependency (pytorch#1302)

ai4co · Jun 21, 2023 · e21e4cf · e21e4cf
1 parent b0e19ff
commit e21e4cf
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -167,9 +167,9 @@ def _main(argv):
     name = args.package_name
     is_nightly = "nightly" in name
     if is_nightly:
-        tensordict_dep = "tensordict>=0.1.1"
-    else:
         tensordict_dep = "tensordict-nightly"
+    else:
+        tensordict_dep = "tensordict>=0.1.1"
 
     if is_nightly:
         version = get_nightly_version()