Commit 1b67fb9 1 parent a3f08ae commit 1b67fb9 Copy full SHA for 1b67fb9
File tree 1 file changed +3
-17
lines changed
1 file changed +3
-17
lines changed Original file line number Diff line number Diff line change @@ -19,24 +19,11 @@ class BaseModelDescription:
19
19
model : str
20
20
sources : ModelSource
21
21
model_file : str
22
- description : str = ""
23
- license : str = ""
24
- size_in_GB : Optional [ float ] = None
22
+ description : str
23
+ license : str
24
+ size_in_GB : float
25
25
additional_files : list [str ] = field (default_factory = list )
26
26
27
- def validate_info (self ) -> None :
28
- if self .license == "" :
29
- raise ValueError ("license is required in builtin model description" )
30
-
31
- if self .description == "" :
32
- raise ValueError ("description is required in builtin model description" )
33
-
34
- if self .size_in_GB is None :
35
- raise ValueError ("size_in_GB is required in builtin model description" )
36
-
37
- def __post_init__ (self ) -> None :
38
- self .validate_info ()
39
-
40
27
41
28
@dataclass (frozen = True )
42
29
class DenseModelDescription (BaseModelDescription ):
@@ -45,7 +32,6 @@ class DenseModelDescription(BaseModelDescription):
45
32
46
33
def __post_init__ (self ) -> None :
47
34
assert self .dim is not None , "dim is required for dense model description"
48
- self .validate_info ()
49
35
50
36
51
37
@dataclass (frozen = True )
You can’t perform that action at this time.
0 commit comments