Fix not handling list of all primitive types in SignatureMixin (#1037)

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>

examples/arena_hard.py

@@ -15,11 +15,12 @@
 import re
 from typing import Any, Dict, List, Optional, Union
 
+from typing_extensions import override
+
 from distilabel.steps import GlobalStep, StepInput
 from distilabel.steps.tasks.base import Task
 from distilabel.steps.tasks.typing import ChatType
 from distilabel.steps.typing import StepOutput
-from typing_extensions import override
 
 
 class ArenaHard(Task):

examples/deepseek_prover.py

@@ -17,14 +17,15 @@
 from textwrap import dedent
 from typing import Any, Dict, List, Optional, Union
 
+from jinja2 import Template
+from pydantic import PrivateAttr
+from typing_extensions import override
+
 from distilabel.llms import InferenceEndpointsLLM
 from distilabel.pipeline import Pipeline
 from distilabel.steps import LoadDataFromHub
 from distilabel.steps.tasks.base import Task
 from distilabel.steps.tasks.typing import ChatType
-from jinja2 import Template
-from pydantic import PrivateAttr
-from typing_extensions import override
 
 _PARSE_DEEPSEEK_PROVER_AUTOFORMAL_REGEX = r"```lean4(.*?)```"
 

examples/structured_generation_with_instructor.py

@@ -14,11 +14,12 @@
 
 from typing import List
 
+from pydantic import BaseModel, Field
+
 from distilabel.llms import MistralLLM
 from distilabel.pipeline import Pipeline
 from distilabel.steps import LoadDataFromDicts
 from distilabel.steps.tasks import TextGeneration
-from pydantic import BaseModel, Field
 
 
 class Node(BaseModel):

examples/structured_generation_with_outlines.py

@@ -15,12 +15,13 @@
 from enum import Enum
 from pathlib import Path
 
+from pydantic import BaseModel, StringConstraints, conint
+from typing_extensions import Annotated
+
 from distilabel.llms import LlamaCppLLM
 from distilabel.pipeline import Pipeline
 from distilabel.steps import LoadDataFromDicts
 from distilabel.steps.tasks import TextGeneration
-from pydantic import BaseModel, StringConstraints, conint
-from typing_extensions import Annotated
 
 
 class Weapon(str, Enum):

src/distilabel/__init__.py

@@ -14,6 +14,6 @@
 
 from rich import traceback as rich_traceback
 
-__version__ = "1.4.0"
+__version__ = "1.4.1"
 
 rich_traceback.install(show_locals=True)

src/distilabel/mixins/signature.py

@@ -67,8 +67,8 @@ def flatten_dump(d: Any, parent_key: str = "", sep: str = "_") -> List:
                 elif isinstance(v, list):
                     if len(v) == 0:
                         items.append((new_key, ""))
-                    elif isinstance(v[0], str):
-                        items.append((new_key, "-".join(v)))
+                    elif isinstance(v[0], (str, float, int, bool)):
+                        items.append((new_key, "-".join(map(str, v))))
                     else:
                         for i, x in enumerate(v):
                             items.extend(flatten_dump(x, f"{new_key}-{i}", sep=sep))