hiro/ray
1
0
Fork 0
mirror of https://github.com/vale981/ray synced 2025-03-06 02:21:39 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

[DataFrame] Implement __array_wrap__ (#2218)

Browse source 
* Implement __array_wrap__

* Removing unnecessary test
This commit is contained in:
Devin Petersohn 2018-06-11 08:56:43 -07:00 • committed by Peter Schafhalter
parent 61139e1509
commit b886ceca47
2 changed files with 2 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
python/ray/dataframe/dataframe.py
View file

@ -4926,9 +4926,8 @@ class DataFrame(object):
return to_pandas(self).__array__(dtype=dtype)
def __array_wrap__(self, result, context=None):
raise NotImplementedError(
"To contribute to Pandas on Ray, please visit "
"github.com/ray-project/ray.")
# TODO: This is very inefficient, see also __array__ and as_matrix
return to_pandas(self).__array_wrap__(result, context=context)
def __getstate__(self):
raise NotImplementedError(

7
python/ray/dataframe/test/test_dataframe.py
View file

@ -3191,13 +3191,6 @@ def test___array__(ray_df, pandas_df):
assert np.array_equal(ray_df.__array__(), pandas_df.__array__())
def test___array_wrap__():
ray_df = create_test_dataframe()
with pytest.raises(NotImplementedError):
ray_df.__array_wrap__(None)
def test___getstate__():
ray_df = create_test_dataframe()