From 8d41f8c652b4d9d6e66b7ff056821c31a0c0f602 Mon Sep 17 00:00:00 2001 From: Silia Taider Date: Wed, 20 May 2026 11:00:17 +0200 Subject: [PATCH] [ML][Python] Fix udnersampling test - rounding below min sampling ratio --- bindings/pyroot/pythonizations/test/ml_dataloader.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/bindings/pyroot/pythonizations/test/ml_dataloader.py b/bindings/pyroot/pythonizations/test/ml_dataloader.py index 748d3517c5d73..c03143dc1c9ab 100644 --- a/bindings/pyroot/pythonizations/test/ml_dataloader.py +++ b/bindings/pyroot/pythonizations/test/ml_dataloader.py @@ -1,3 +1,4 @@ +import math import os import unittest from random import randrange, uniform @@ -5779,7 +5780,7 @@ def test14_big_data_replacement_false(self): entries_in_rdf_minor = randrange(8000, 9999) batch_size = randrange(100, 501) min_allowed_sampling_ratio = entries_in_rdf_minor / entries_in_rdf_major - sampling_ratio = round(uniform(min_allowed_sampling_ratio, 2), 2) + sampling_ratio = math.ceil(uniform(min_allowed_sampling_ratio, 2) * 100) / 100 error_message = f"\n Batch size: {batch_size}\ Number of major entries: {entries_in_rdf_major} \