Ignore disable_cuda_device_placement attribute for pipeline signature

argilla-io · Aug 1, 2024 · d0fbd5d · d0fbd5d
1 parent c776d36
commit d0fbd5d
Showing 1 changed file with 4 additions and 3 deletions.
diff --git 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'."