davidberenstein1957 HF staff commited on
Commit
d2df8be
Β·
1 Parent(s): 342e737

remove obsolete class

Browse files
src/synthetic_dataset_generator/pipelines/textcat.py CHANGED
@@ -60,19 +60,6 @@ class TextClassificationTask(BaseModel):
60
  )
61
 
62
 
63
- class DatasetDescription(BaseModel):
64
- description: str = Field(
65
- ...,
66
- title="description",
67
- description="The description of the dataset.",
68
- )
69
- labels: list[str] = Field(
70
- ...,
71
- title="labels",
72
- description="The possible labels for the classification task.",
73
- )
74
-
75
-
76
  def get_prompt_generator():
77
  prompt_generator = TextGeneration(
78
  llm=InferenceEndpointsLLM(
 
60
  )
61
 
62
 
 
 
 
 
 
 
 
 
 
 
 
 
 
63
  def get_prompt_generator():
64
  prompt_generator = TextGeneration(
65
  llm=InferenceEndpointsLLM(