fix: floating point issue with python not returning 1.0 for prob sum (#261)

Co-authored-by: Raktim Mitra <raktim@localhost>

src/modelhub/utils/datasets.py

@@ -189,7 +189,8 @@ def recursively_instantiate_datasets_and_samplers(
 
         # ... check that the sum of probabilities of all datasets is 1
         assert (
-            sum(dataset_info["probability"] for dataset_info in datasets_info) == 1.0
+            abs(1 - sum(dataset_info["probability"] for dataset_info in datasets_info))
+            < 1e-5
         ), "Sum of probabilities must be 1.0"
 
         # ... compose the list of datasets into a single dataset