From e98e43c5fa483558b72c96bd843bae7386d61af5 Mon Sep 17 00:00:00 2001
From: rjzamora <rzamora217@gmail.com>
Date: Thu, 4 Apr 2024 08:05:55 -0700
Subject: [PATCH] use the proper timeseries version

---
 dask/datasets.py | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

diff --git a/dask/datasets.py b/dask/datasets.py
index c2684515bda..89d6f876ad5 100644
--- a/dask/datasets.py
+++ b/dask/datasets.py
@@ -55,6 +55,24 @@ def timeseries(
     ...     id_lam=1000  # control number of items in id column
     ... )
     """
+    from dask.dataframe import _dask_expr_enabled
+
+    if _dask_expr_enabled():
+        try:
+            from dask_expr.datasets import timeseries as _timeseries
+
+            return _timeseries(
+                start=start,
+                end=end,
+                freq=freq,
+                partition_freq=partition_freq,
+                dtypes=dtypes,
+                seed=seed,
+                **kwargs,
+            )
+        except ImportError:
+            pass
+
     from dask.dataframe.io.demo import make_timeseries
 
     if dtypes is None: