fix for issue h2oai#15947

Devanshusisodiya · Feb 13, 2024 · af62fa4 · af62fa4
1 parent b328502
commit af62fa4
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/h2o-py/h2o/frame.py b/h2o-py/h2o/frame.py
@@ -146,6 +146,12 @@ def _upload_python_object(self, python_obj, destination_frame=None, header=0, se
         if not column_names:
             column_names = col_header
 
+        if skipped_columns:
+            column_names = [column_name for column_name in column_names[:len(column_names)-len(skipped_columns)]]
+
+        if not column_names:
+            raise H2OValueError("skipped_columns cannot contain all columns")
+
         # create a temporary file that will be written to
         tmp_handle, tmp_path = tempfile.mkstemp(suffix=".csv")
         tmp_file = os.fdopen(tmp_handle, 'w', **H2OFrame.__fdopen_kwargs)