Nicer read_parquet prefix (#998)

Co-authored-by: Patrick Hoefler <[email protected]>
dask · Mar 25, 2024 · cee1ba3 · cee1ba3
1 parent df0f694
commit cee1ba3
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 6 deletions.
diff --git a/dask_expr/_core.py b/dask_expr/_core.py
@@ -442,11 +442,13 @@ def lower_completely(self) -> Expr:
     def _lower(self):
         return
 
+    @functools.cached_property
+    def _funcname(self):
+        return funcname(type(self)).lower()
+
     @functools.cached_property
     def _name(self):
-        return (
-            funcname(type(self)).lower() + "-" + _tokenize_deterministic(*self.operands)
-        )
+        return self._funcname + "-" + _tokenize_deterministic(*self.operands)
 
     @property
     def _meta(self):

diff --git a/dask_expr/io/io.py b/dask_expr/io/io.py
@@ -103,7 +103,7 @@ class FusedIO(BlockwiseIO):
     @functools.cached_property
     def _name(self):
         return (
-            funcname(type(self.operand("_expr"))).lower()
+            self.operand("_expr")._funcname
             + "-fused-"
             + _tokenize_deterministic(*self.operands)
         )

diff --git a/dask_expr/io/parquet.py b/dask_expr/io/parquet.py
@@ -631,12 +631,18 @@ def columns(self):
         else:
             return _convert_to_list(columns_operand)
 
+    @cached_property
+    def _funcname(self):
+        return "read_parquet"
+
     @cached_property
     def _name(self):
         return (
-            funcname(type(self)).lower()
+            self._funcname
             + "-"
-            + _tokenize_deterministic(self.checksum, *self.operands[:-1])
+            + _tokenize_deterministic(
+                funcname(type(self)), self.checksum, *self.operands[:-1]
+            )
         )
 
     @property