FIX-#4058: Allow pickling empty dataframes and series. (#4095)

Co-authored-by: Rehan Sohail Durrani <[email protected]> Signed-off-by: mvashishtha <[email protected]>
modin-project · Feb 4, 2022 · 025190c · 025190c
1 parent e2996e8
commit 025190c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/modin/pandas/base.py b/modin/pandas/base.py
@@ -91,6 +91,7 @@
     "_inflate_light",
     "_inflate_full",
     "__reduce__",
+    "__reduce_ex__",
 } | _ATTRS_NO_LOOKUP
 
 

diff --git a/modin/pandas/test/dataframe/test_pickle.py b/modin/pandas/test/dataframe/test_pickle.py
@@ -39,6 +39,9 @@ def persistent(request):
     PersistentPickle.put(old)
 
 
+@pytest.mark.parametrize(
+    "modin_df", [pytest.param(modin_df), pytest.param(pd.DataFrame(), id="empty_df")]
+)
 def test_dataframe_pickle(modin_df, persistent):
     other = pickle.loads(pickle.dumps(modin_df))
     df_equals(modin_df, other)