Update fslite/fs/ml.py

ypriverol · qodo-merge-pro[bot] · web-flow · commit 86081174f313 · 2024-09-25T13:26:40.000+01:00
Co-authored-by: codiumai-pr-agent-pro[bot] &lt;151058649+codiumai-pr-agent-pro[bot]@users.noreply.github.com&gt;
diff --git a/fslite/fs/ml.py b/fslite/fs/ml.py
@@ -140,7 +140,7 @@ def select_features(self, fsdf: FSDataFrame) -> FSDataFrame:
         """
 
         if fsdf is None or fsdf.count_features() == 0 or fsdf.count_instances() == 0:
-            raise ValueError(
+            raise InvalidDataError(
                 "The data frame is empty or does not contain any features."
             )
 

Original file line number	Diff line number	Diff line change
`@@ -140,7 +140,7 @@ def select_features(self, fsdf: FSDataFrame) -> FSDataFrame:`
`140`	`140`	`"""`
`141`	`141`
`142`	`142`	`if fsdf is None or fsdf.count_features() == 0 or fsdf.count_instances() == 0:`
`143`		`- raise ValueError(`
	`143`	`+ raise InvalidDataError(`
`144`	`144`	`"The data frame is empty or does not contain any features."`
`145`	`145`	`)`
`146`	`146`