[BugFix] Fix Ray collector on Python > 3.8 (pytorch#2015)

ai4co · Mar 18, 2024 · 87f3437 · 87f3437
1 parent 77d2fc9
commit 87f3437
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/torchrl/collectors/collectors.py b/torchrl/collectors/collectors.py
@@ -235,6 +235,9 @@ def __repr__(self) -> str:
         string = f"{self.__class__.__name__}()"
         return string
 
+    def __class_getitem__(self, index):
+        raise NotImplementedError
+
 
 @accept_remote_rref_udf_invocation
 class SyncDataCollector(DataCollectorBase):