fix: yaml parsing converts floats like 1.0 into integer 1 causing assertion failure

Signed-off-by: Harikrishnan Balagopal <harikrishmenon@gmail.com>

Author: HarikrishnanBalagopal

tuning/data/data_config.py

@@ -100,6 +100,9 @@ def _validate_dataset_config(dataset_config) -> DataSetConfig:
         c.builder = builder
     if "sampling" in kwargs and kwargs["sampling"] is not None:
         ratio = kwargs["sampling"]
+        if isinstance(ratio, int):
+            # YAML parsing causes 1.0 (float) to be cast to 1 (int) causing assertion error
+            ratio = float(ratio)
         assert isinstance(ratio, float) and (
             0 <= ratio <= 1.0
         ), f"sampling ratio: {ratio} should be float and in range [0.0,1.0]"