Fix

Signed-off-by: Igoshev, Iaroslav <[email protected]>
YarShev · Apr 10, 2024 · b1357f3 · b1357f3
1 parent 717a328
commit b1357f3
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/modin/core/storage_formats/pandas/query_compiler.py b/modin/core/storage_formats/pandas/query_compiler.py
@@ -2908,7 +2908,7 @@ def getitem_row_array(self, key):
         )
 
     def setitem(self, axis, key, value):
-        if axis == 1:
+        if axis == 0:
             value = self._wrap_column_data(value)
         return self._setitem(axis=axis, key=key, value=value, how=None)
 

diff --git a/modin/tests/pandas/dataframe/test_indexing.py b/modin/tests/pandas/dataframe/test_indexing.py
@@ -565,6 +565,18 @@ def test_loc_5829():
     )
 
 
+def test_loc_7135():
+    data = np.random.randint(0, 100, size=(2**16, 2**8))
+    modin_df, pandas_df = create_test_dfs(data)
+    key = len(pandas_df) - 1
+    eval_loc(
+        modin_df,
+        pandas_df,
+        value=list(range(2**8)),
+        key=key,
+    )
+
+
 # This tests the bug from https://github.com/modin-project/modin/issues/3736
 def test_loc_setting_single_categorical_column():
     modin_df = pd.DataFrame({"status": ["a", "b", "c"]}, dtype="category")