ENH: Fix DataFrame._data deprecated warnings (#793)

Co-authored-by: mergify[bot] <37929162+mergify[bot]@users.noreply.github.com>
xorbitsai · Aug 19, 2024 · 2991976 · 2991976
1 parent c4e4931
commit 2991976
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/CI/test_basic_execution.py b/CI/test_basic_execution.py
@@ -7,10 +7,9 @@
 
 
 def test_basic_cases():
- with warnings.catch_warnings():
- # TODO: change warnings.simplefilter("error") to "module"
+ with warnings.catch_warnings(): 
  # "error" help us find the deprecated APIs
- warnings.simplefilter("module")
+ warnings.simplefilter("error")
  import xorbits
  import xorbits.pandas as pd
  import xorbits.numpy as np

diff --git a/python/xorbits/_mars/_utils.pyx b/python/xorbits/_mars/_utils.pyx
@@ -201,7 +201,7 @@ cdef list tokenize_pandas_series(ob):
 
 
 cdef list tokenize_pandas_dataframe(ob):
- l = [block.values for block in ob._data.blocks]
+ l = [block.values for block in ob._mgr.blocks]
  l.extend([ob.columns, ob.index])
  return iterative_tokenize(l)