diff --git a/nemo/utils/model_utils.py b/nemo/utils/model_utils.py index 845c6bdf9953..b2a6abbf54aa 100644 --- a/nemo/utils/model_utils.py +++ b/nemo/utils/model_utils.py @@ -54,10 +54,10 @@ class ArtifactPathType(Enum): TAR_PATH = 1 -@dataclass(init=False) +@dataclass class ArtifactItem: - path: str - path_type: ArtifactPathType + path: str = "" + path_type: ArtifactPathType = ArtifactPathType.LOCAL_PATH hashed_path: Optional[str] = None