Danny Liu commited on
Commit
f45905d
·
1 Parent(s): 1b4fc60

fix src/display/utils.py bug

Browse files
Files changed (1) hide show
  1. src/display/utils.py +2 -2
src/display/utils.py CHANGED
@@ -12,7 +12,7 @@ def fields(raw_class):
12
  # These classes are for user facing column names,
13
  # to avoid having to change them all around the code
14
  # when a modif is needed
15
- @dataclass
16
  class ColumnContent:
17
  name: str
18
  type: str
@@ -37,7 +37,7 @@ auto_eval_column_dict.append(["license", ColumnContent, ColumnContent("Hub Licen
37
  AutoEvalColumn = make_dataclass("AutoEvalColumn", auto_eval_column_dict, frozen=True)
38
 
39
  ## All the model information that we might need
40
- @dataclass
41
  class ModelDetails:
42
  name: str
43
  display_name: str = ""
 
12
  # These classes are for user facing column names,
13
  # to avoid having to change them all around the code
14
  # when a modif is needed
15
+ @dataclass(frozen=True)
16
  class ColumnContent:
17
  name: str
18
  type: str
 
37
  AutoEvalColumn = make_dataclass("AutoEvalColumn", auto_eval_column_dict, frozen=True)
38
 
39
  ## All the model information that we might need
40
+ @dataclass(frozen=True)
41
  class ModelDetails:
42
  name: str
43
  display_name: str = ""