diff --git a/requirements/cli.txt b/requirements/cli.txt index b43f09ad..c2676ed8 100644 --- a/requirements/cli.txt +++ b/requirements/cli.txt @@ -1,5 +1,5 @@ SQLAlchemy>=1.1.1 -tabulate==0.8.2 +tabulate==0.9.0 xhtml2pdf==0.2.15 ohio>=0.2.0 markdown2==2.3.5 diff --git a/requirements/main.txt b/requirements/main.txt index cad49c51..11357c2c 100644 --- a/requirements/main.txt +++ b/requirements/main.txt @@ -12,5 +12,5 @@ fairlearn>=0.8.0 hydra-core>=1.3.0 validators>=0.22.0 hyperparameter-tuning>=0.3.1 -numpy==1.23.5 -fastparquet==2024.2.0 \ No newline at end of file +numpy>=1.23.5 +fastparquet==2024.2.0 diff --git a/tests/test_generic.py b/tests/test_generic.py index 8653ba2f..00a53121 100644 --- a/tests/test_generic.py +++ b/tests/test_generic.py @@ -199,11 +199,10 @@ def test_invalid_splits_warn(self): ), split_values={"train": 0.3, "validation": 0.1, "test": 0.2}, ) - self.assertEqual( - cm.output, - [ - "WARNING:datasets.GenericDataset:Using only 0.6000000000000001 of the dataset." - ], + self.assertEqual(len(cm.output), 1) + self.assertIn( + "WARNING:datasets.GenericDataset:Using only 0.6", + cm.output[0] ) def test_missing_splits_column(self):