Cleanup

Author: cameronwaterman

nisystemlink/clients/dataframe/_data_frame_client.py

@@ -1,12 +1,9 @@
 """Implementation of DataFrameClient."""
 
-try:  # Optional pyarrow dependency
-    import pyarrow as pa  # type: ignore
-except Exception:  # pragma: no cover - pyarrow not installed
-    pa = None  # type: ignore
+import pyarrow as pa  # type: ignore
 from collections.abc import Iterable
 from io import BytesIO
-from typing import Any, List, Optional, Union
+from typing import List, Optional, Union
 
 from nisystemlink.clients import core
 from nisystemlink.clients.core._uplink._base_client import BaseClient
@@ -19,7 +16,7 @@
 )
 from nisystemlink.clients.core.helpers import IteratorFileLike
 from requests.models import Response
-from uplink import Body, Field, Path, Query, headers, retry
+from uplink import Body, Field, Path, Query, retry
 
 from . import models
 
@@ -452,7 +449,7 @@ def query_decimated_data(
         """
         ...
 
-    def _iter_content_filelike_wrapper(response: Response) -> IteratorFileLike:  # type: ignore
+    def _iter_content_filelike_wrapper(response: Response) -> IteratorFileLike:
         return IteratorFileLike(response.iter_content(chunk_size=4096))
 
     @response_handler(_iter_content_filelike_wrapper)

tests/integration/dataframe/test_dataframe.py

@@ -1,6 +1,6 @@
 # -*- coding: utf-8 -*-
 from datetime import datetime, timezone
-from typing import List, Optional, Iterable
+from typing import List, Optional
 
 import pytest  # type: ignore
 import responses