linting progress

Author: dougbrn

src/nested_pandas/nestedframe/core.py

@@ -1082,8 +1082,9 @@ def to_parquet(self, path, **kwargs) -> None:
         path : str
             The path to the parquet file
         kwargs : keyword arguments, optional
-            Keyword arguments to pass to the function
-            `pyarrow.parquet.write_table <https://arrow.apache.org/docs/python/generated/pyarrow.parquet.write_table.html>`_
+            Keyword arguments to pass to
+            `pyarrow.parquet.write_table
+            <https://arrow.apache.org/docs/python/generated/pyarrow.parquet.write_table.html>`_
 
         Returns
         -------

src/nested_pandas/nestedframe/io.py

@@ -66,8 +66,8 @@ def read_parquet(
 
     Partial loading:
 
-    >>> #Load only the "a" sub-column of the "nested" column
-    >>> nf = npd.read_parquet("path/to/file.parquet", columns=["nested.a"])
+    >>> #Load only the "flux" sub-column of the "nested" column
+    >>> nf = npd.read_parquet("path/to/file.parquet", columns=["a", "nested.flux"])
     """
 
     # Type convergence for reject_nesting

tests/nested_pandas/nestedframe/test_io.py

@@ -2,12 +2,11 @@
 import tempfile
 
 import pandas as pd
+import pyarrow as pa
 import pytest
-from nested_pandas import NestedFrame, read_parquet
+from nested_pandas import read_parquet
 from pandas.testing import assert_frame_equal
 
-import pyarrow as pa
-
 
 def test_read_parquet():
     """Test reading a parquet file with no columns specified"""
@@ -25,12 +24,16 @@ def test_read_parquet():
     assert nf.lincc.nest.fields == ["band", "frameworks"]
 
 
-@pytest.mark.parametrize("columns", [["a", "flux"],
-                                     ["flux", "nested", "lincc"],
-                                     ["nested.flux", "nested.band"],
-                                     ["flux", "nested.flux"],
-                                     ["nested.band", "lincc.band"],
-                                     ])
+@pytest.mark.parametrize(
+    "columns",
+    [
+        ["a", "flux"],
+        ["flux", "nested", "lincc"],
+        ["nested.flux", "nested.band"],
+        ["flux", "nested.flux"],
+        ["nested.band", "lincc.band"],
+    ],
+)
 def test_read_parquet_column_selection(columns):
     """Test reading a parquet file with column selection"""
     # Load in the example file
@@ -58,13 +61,12 @@ def test_read_parquet_column_selection(columns):
         assert nf.nested.nest.fields == ["band"]
         assert nf.lincc.nest.fields == ["band"]
 
+
 @pytest.mark.parametrize("reject", [["nested"], "nested"])
 def test_read_parquet_reject_nesting(reject):
     """Test reading a parquet file with column selection"""
     # Load in the example file
-    nf = read_parquet("tests/test_data/nested.parquet",
-                      columns=["a", "nested"],
-                      reject_nesting=reject)
+    nf = read_parquet("tests/test_data/nested.parquet", columns=["a", "nested"], reject_nesting=reject)
 
     # Check the columns
     assert nf.columns.tolist() == ["a", "nested"]
@@ -78,9 +80,7 @@ def test_read_parquet_reject_nesting(reject):
 def test_read_parquet_reject_nesting_partial_loading():
     """Test reading a parquet file with column selection"""
     # Load in the example file
-    nf = read_parquet("tests/test_data/nested.parquet",
-                      columns=["a", "nested.t"],
-                      reject_nesting=["nested"])
+    nf = read_parquet("tests/test_data/nested.parquet", columns=["a", "nested.t"], reject_nesting=["nested"])
 
     # Check the columns
     assert nf.columns.tolist() == ["a", "t"]