Fix lint

microsoft · Aug 5, 2023 · c21fab4 · c21fab4
1 parent 570ca64
commit c21fab4
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/python-package/lightgbm/arrow.py b/python-package/lightgbm/arrow.py
@@ -1,3 +1,5 @@
+# coding: utf-8
+"""Utilities for handling Arrow in LightGBM."""
 from contextlib import contextmanager
 from dataclasses import dataclass
 from typing import Iterator, Union
@@ -16,10 +18,12 @@ class ArrowCArray:
 
     @property
     def chunks_ptr(self) -> int:
+        """Returns the address of the pointer to the list of chunks making up the array."""
         return int(ffi.cast("uintptr_t", ffi.addressof(self.chunks[0])))
 
     @property
     def schema_ptr(self) -> int:
+        """Returns the address of the pointer to the schema of the array."""
         return int(ffi.cast("uintptr_t", self.schema))
 
 

diff --git a/tests/python_package_test/test_arrow.py b/tests/python_package_test/test_arrow.py
@@ -6,7 +6,6 @@
 
 import lightgbm as lgb
 import numpy as np
-import pandas as pd
 import pyarrow as pa
 import pytest