From a80d224c581a47974d034e4a3c0f1fb3e4d04d00 Mon Sep 17 00:00:00 2001
From: skshetry <18718008+skshetry@users.noreply.github.com>
Date: Thu, 14 Mar 2024 15:20:17 +0545
Subject: [PATCH] fix dvcx import (#10351)

---
 dvc/repo/datasets.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/dvc/repo/datasets.py b/dvc/repo/datasets.py
index c06308f49..992f02f59 100644
--- a/dvc/repo/datasets.py
+++ b/dvc/repo/datasets.py
@@ -16,8 +16,7 @@ from dvc.types import StrPath
 from dvc_data.hashfile.meta import Meta
 
 if TYPE_CHECKING:
-    from dql.dataset import DatasetRecord  # type: ignore[import]
-    from dvcx.dataset import DatasetVersion  # type: ignore[import]
+    from dvcx.dataset import DatasetRecord, DatasetVersion  # type: ignore[import]
     from typing_extensions import Self
 
     from dvc.repo import Repo