mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-02 07:18:25 +01:00
skshetry
b647475a64
* drop Python 3.8 Note that this PR only makes typing related changes. Other changes can be done once this gets merged, they are relatively few in numbers. * bump build python version * add typing_extensions to lint group
53 lines
1.4 KiB
Python
53 lines
1.4 KiB
Python
from dataclasses import asdict, dataclass, field, fields
|
|
from typing import Any, ClassVar, Optional
|
|
|
|
from funcy import compact
|
|
from voluptuous import Required
|
|
|
|
|
|
@dataclass
|
|
class Annotation:
|
|
PARAM_DESC: ClassVar[str] = "desc"
|
|
PARAM_TYPE: ClassVar[str] = "type"
|
|
PARAM_LABELS: ClassVar[str] = "labels"
|
|
PARAM_META: ClassVar[str] = "meta"
|
|
|
|
desc: Optional[str] = None
|
|
type: Optional[str] = None
|
|
labels: list[str] = field(default_factory=list)
|
|
meta: dict[str, Any] = field(default_factory=dict)
|
|
|
|
def to_dict(self) -> dict[str, str]:
|
|
return compact(asdict(self))
|
|
|
|
|
|
@dataclass
|
|
class Artifact:
|
|
PARAM_PATH: ClassVar[str] = "path"
|
|
PARAM_DESC: ClassVar[str] = "desc"
|
|
PARAM_TYPE: ClassVar[str] = "type"
|
|
PARAM_LABELS: ClassVar[str] = "labels"
|
|
PARAM_META: ClassVar[str] = "meta"
|
|
|
|
path: str
|
|
desc: Optional[str] = None
|
|
type: Optional[str] = None
|
|
labels: list[str] = field(default_factory=list)
|
|
meta: dict[str, Any] = field(default_factory=dict)
|
|
|
|
def to_dict(self) -> dict[str, str]:
|
|
return compact(asdict(self))
|
|
|
|
|
|
ANNOTATION_FIELDS = [field.name for field in fields(Annotation)]
|
|
ANNOTATION_SCHEMA = {
|
|
Annotation.PARAM_DESC: str,
|
|
Annotation.PARAM_TYPE: str,
|
|
Annotation.PARAM_LABELS: [str],
|
|
Annotation.PARAM_META: object,
|
|
}
|
|
ARTIFACT_SCHEMA: dict[Any, Any] = {
|
|
Required(Artifact.PARAM_PATH): str,
|
|
**ANNOTATION_SCHEMA, # type: ignore[arg-type]
|
|
}
|