lazily import

iterative · Feb 21, 2024 · 6df9f0e · 6df9f0e
1 parent 0b3a065
commit 6df9f0e
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/dvc/commands/dataset.py b/dvc/commands/dataset.py
@@ -4,7 +4,6 @@
 from dvc.cli.command import CmdBase
 from dvc.cli.utils import append_doc_link
 from dvc.log import logger
-from dvc.repo.datasets import URLDatasetLock
 
 if TYPE_CHECKING:
     from rich.text import Text
@@ -74,7 +73,7 @@ def run(self):
 class CmdDatasetUpdate(CmdBase):
     def display(self, dataset: "Dataset", new: "Dataset"):
         from dvc.commands.checkout import log_changes
-        from dvc.repo.datasets import DVCDatasetLock, DVCXDatasetLock
+        from dvc.repo.datasets import DVCDatasetLock, DVCXDatasetLock, URLDatasetLock
         from dvc.ui import ui
 
         if not dataset.lock: