diff --git a/src/distilabel/pipeline/base.py b/src/distilabel/pipeline/base.py index 42a8f871e2..237bfa5c8c 100644 --- a/src/distilabel/pipeline/base.py +++ b/src/distilabel/pipeline/base.py @@ -263,9 +263,10 @@ def _create_signature(self) -> str: elif isinstance(value, (list, tuple)): # runtime_parameters_info step_info += "-".join([str(v) for v in value]) - elif isinstance(value, (int, str, float)): - # batch_size/name - step_info += str(value) + elif isinstance(value, (int, str, float, bool)): + if argument != "disable_cuda_device_placement": + # batch_size/name + step_info += str(value) else: raise ValueError( f"Field '{argument}' in step '{step['name']}' has type {type(value)}, explicitly cast the type to 'str'."