[Data] Record more telemetry for newly added datasources (#45647)

This PR is to add the telemetry recording for newly added datasources. Signed-off-by: Cheng Su <[email protected]>
ray-project · May 31, 2024 · 3c9edf1 · 3c9edf1
1 parent a30630a
commit 3c9edf1
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/python/ray/data/_internal/logical/util.py b/python/ray/data/_internal/logical/util.py
@@ -28,8 +28,14 @@
     "ReadNumpy",
     "ReadTFRecord",
     "ReadBinary",
-    "ReadCustom",
     "ReadTorch",
+    "ReadAvro",
+    "ReadWebDataset",
+    "ReadSQL",
+    "ReadDatabricksUC",
+    "ReadLance",
+    "ReadHuggingFace",
+    "ReadCustom",
     # From
     "FromArrow",
     "FromItems",
@@ -43,6 +49,8 @@
     "WriteTFRecord",
     "WriteNumpy",
     "WriteMongo",
+    "WriteWebDataset",
+    "WriteSQL",
     "WriteCustom",
     # Map
     "Map",