support Annotated types

iterative · Jan 9, 2024 · fcddfd6 · fcddfd6
1 parent 4e0186a
commit fcddfd6
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/dvc/api/dataset.py b/dvc/api/dataset.py
@@ -21,6 +21,12 @@ def _get_datasets() -> Dict[str, Dict[str, Any]]:
         return repo.index.datasets
 
 
+def _get_raw(name: str) -> Dict[str, Any]:
+    return _get_datasets()[name]
+
+
 def get(cls: "Type[T]", name: str) -> "T":
-    datasets = _get_datasets()
-    return cls.model_validate(datasets[name])
+    from pydantic import TypeAdapter
+
+    d = _get_raw(name)
+    return TypeAdapter(cls).validate_python(d)