first entries in rl4co.data.__init__

ngastzepeda · Nov 27, 2024 · fe67119 · fe67119
1 parent ec76438
commit fe67119
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/rl4co/data/__init__.py b/rl4co/data/__init__.py
@@ -0,0 +1,2 @@
+from rl4co.data.dataset import FastTdDataset, TensorDictDataset
+from rl4co.data.utils import save_tensordict_to_npz