thanhnt-cf commited on
Commit
98ef152
·
1 Parent(s): b652f9c

fix issues in description

Browse files
Files changed (1) hide show
  1. app/services/base.py +5 -4
app/services/base.py CHANGED
@@ -27,7 +27,7 @@ def cf_style_to_pydantic_percentage_shema(
27
  else:
28
  multiple = False
29
  class_name = "Class_" + attribute.capitalize()
30
- multiple_desc = "multi-label classification" if multiple else "classification"
31
  attribute_desc = attribute_info.description
32
  attribute_line = f'{attribute}: {class_name} = Field("", description="{multiple_desc}, {attribute_desc}")'
33
 
@@ -146,24 +146,25 @@ class BaseAttributionService(ABC):
146
  init_reevaluate_data = {}
147
  for field_name, field in attributes_percentage_model.model_fields.items(): # type: ignore
148
  print(f"{field_name}: {field.description}")
149
- if "single value" in field.description.lower():
150
  max_percentage = 0
151
  for k, v in reevaluate_data[field_name].items():
152
  if v > max_percentage:
153
  max_percentage = v
154
  init_reevaluate_data[field_name] = k
155
- elif "multiple values" in field.description.lower():
156
  init_list = []
157
  for k, v in reevaluate_data[field_name].items():
158
  if v >= 60:
159
  init_list.append(k)
160
  init_reevaluate_data[field_name] = init_list
 
 
161
 
162
  # reverse the key mapping to the original keys
163
  reverse_data = {}
164
  for key, value in init_reevaluate_data.items():
165
  reverse_data[reverse_mapping[key]] = value
166
-
167
  return data, reverse_data
168
 
169
  async def follow_schema_with_validation(
 
27
  else:
28
  multiple = False
29
  class_name = "Class_" + attribute.capitalize()
30
+ multiple_desc = "multi-label classification" if multiple else "single-label classification"
31
  attribute_desc = attribute_info.description
32
  attribute_line = f'{attribute}: {class_name} = Field("", description="{multiple_desc}, {attribute_desc}")'
33
 
 
146
  init_reevaluate_data = {}
147
  for field_name, field in attributes_percentage_model.model_fields.items(): # type: ignore
148
  print(f"{field_name}: {field.description}")
149
+ if "single-label" in field.description.lower():
150
  max_percentage = 0
151
  for k, v in reevaluate_data[field_name].items():
152
  if v > max_percentage:
153
  max_percentage = v
154
  init_reevaluate_data[field_name] = k
155
+ elif "multi-label" in field.description.lower():
156
  init_list = []
157
  for k, v in reevaluate_data[field_name].items():
158
  if v >= 60:
159
  init_list.append(k)
160
  init_reevaluate_data[field_name] = init_list
161
+ else:
162
+ assert False, f"The description does not contain 'single-label' or 'multi-label': {field.description}"
163
 
164
  # reverse the key mapping to the original keys
165
  reverse_data = {}
166
  for key, value in init_reevaluate_data.items():
167
  reverse_data[reverse_mapping[key]] = value
 
168
  return data, reverse_data
169
 
170
  async def follow_schema_with_validation(