decorate return_data for TNGCluster

cbyrohl · Jul 11, 2023 · bd31a14 · bd31a14
1 parent 8393f94
commit bd31a14
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/scida/customs/arepo/TNGcluster/dataset.py b/src/scida/customs/arepo/TNGcluster/dataset.py
@@ -72,6 +72,10 @@ def __init__(self, *args, **kwargs):
         ]
         pass
 
+    @TNGClusterSelector()
+    def return_data(self):
+        return super().return_data()
+
     @classmethod
     def validate_path(
         cls, path: Union[str, os.PathLike], *args, **kwargs