Review comments

rapidsai · Aug 25, 2021 · 61bc39d · 61bc39d
1 parent bd9be97
commit 61bc39d
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/python/dask_cudf/dask_cudf/accessors.py b/python/dask_cudf/dask_cudf/accessors.py
@@ -47,8 +47,8 @@ def explode(self):
 
         Examples
         --------
-        >>> import cudf, dask_cudf as dgd
-        >>> ds = dgd.from_cudf(cudf.Series(
+        >>> import cudf, dask_cudf
+        >>> ds = dask_cudf.from_cudf(cudf.Series(
         ...     [{'a': 42, 'b': 'str1', 'c': [-1]},
         ...      {'a': 0,  'b': 'str2', 'c': [400, 500]},
         ...      {'a': 7,  'b': '',     'c': []}]), npartitions=2)

diff --git a/python/dask_cudf/dask_cudf/tests/test_accessor.py b/python/dask_cudf/dask_cudf/tests/test_accessor.py
@@ -511,7 +511,7 @@ def test_dask_struct_field_Int_Error(data):
     ],
 )
 def test_struct_explode(data):
-    expect = pd.DataFrame(data)
-    got = dgd.from_cudf(Series(data), 2).struct.explode().compute()
+    expect = Series(data).struct.explode()
+    got = dgd.from_cudf(Series(data), 2).struct.explode()
 
-    assert_eq(expect, got)
+    assert_eq(expect, got.compute())