diff --git a/db_dtypes/__init__.py b/db_dtypes/__init__.py index 54721a3..ad4ea33 100644 --- a/db_dtypes/__init__.py +++ b/db_dtypes/__init__.py @@ -27,10 +27,8 @@ import pyarrow import pyarrow.compute - -from db_dtypes.version import __version__ from db_dtypes import core - +from db_dtypes.version import __version__ date_dtype_name = "dbdate" time_dtype_name = "dbtime" diff --git a/db_dtypes/core.py b/db_dtypes/core.py index 567b931..7c9eb6b 100644 --- a/db_dtypes/core.py +++ b/db_dtypes/core.py @@ -21,7 +21,6 @@ from db_dtypes import pandas_backports - pandas_release = pandas_backports.pandas_release diff --git a/db_dtypes/pandas_backports.py b/db_dtypes/pandas_backports.py index 0966e83..e3aea17 100644 --- a/db_dtypes/pandas_backports.py +++ b/db_dtypes/pandas_backports.py @@ -29,7 +29,6 @@ import pandas.compat.numpy.function import pandas.core.nanops - pandas_release = packaging.version.parse(pandas.__version__).release # Create aliases for private methods in case they move in a future version. diff --git a/noxfile.py b/noxfile.py index 102670a..d6735da 100644 --- a/noxfile.py +++ b/noxfile.py @@ -21,7 +21,6 @@ import os import pathlib import re -import re import shutil from typing import Dict, List import warnings diff --git a/samples/snippets/pandas_date_and_time.py b/samples/snippets/pandas_date_and_time.py index 3292e6c..b6e5581 100644 --- a/samples/snippets/pandas_date_and_time.py +++ b/samples/snippets/pandas_date_and_time.py @@ -17,7 +17,9 @@ def pandas_date_and_time(): # [START bigquery_pandas_date_create] import datetime + import pandas as pd + import db_dtypes # noqa import to register dtypes dates = pd.Series([datetime.date(2021, 9, 17), "2021-9-18"], dtype="dbdate") diff --git a/setup.py b/setup.py index e91b109..276a940 100644 --- a/setup.py +++ b/setup.py @@ -15,6 +15,7 @@ import io import os import re + from setuptools import setup # Package metadata. diff --git a/tests/compliance/date/test_date_compliance.py b/tests/compliance/date/test_date_compliance.py index 5c43287..038005a 100644 --- a/tests/compliance/date/test_date_compliance.py +++ b/tests/compliance/date/test_date_compliance.py @@ -21,13 +21,12 @@ """ import pandas -from pandas.tests.extension import base import pandas._testing as tm +from pandas.tests.extension import base import pytest import db_dtypes - # TODO(https://github.com/googleapis/python-db-dtypes-pandas/issues/87): Add # compliance tests for arithmetic operations. diff --git a/tests/compliance/time/test_time_compliance.py b/tests/compliance/time/test_time_compliance.py index 3b36d16..f894ba5 100644 --- a/tests/compliance/time/test_time_compliance.py +++ b/tests/compliance/time/test_time_compliance.py @@ -21,13 +21,12 @@ """ import pandas -from pandas.tests.extension import base import pandas._testing as tm +from pandas.tests.extension import base import pytest import db_dtypes - # TODO(https://github.com/googleapis/python-db-dtypes-pandas/issues/87): Add # compliance tests for arithmetic operations. diff --git a/tests/unit/test_arrow.py b/tests/unit/test_arrow.py index 56bbd01..a7a38a2 100644 --- a/tests/unit/test_arrow.py +++ b/tests/unit/test_arrow.py @@ -23,7 +23,6 @@ import db_dtypes - SECOND_NANOS = 1_000_000_000 MINUTE_NANOS = 60 * SECOND_NANOS HOUR_NANOS = 60 * MINUTE_NANOS diff --git a/tests/unit/test_date.py b/tests/unit/test_date.py index fddf1a0..9dbc6d9 100644 --- a/tests/unit/test_date.py +++ b/tests/unit/test_date.py @@ -25,7 +25,6 @@ import db_dtypes from db_dtypes import pandas_backports - VALUE_PARSING_TEST_CASES = [ # Min/Max values for pandas.Timestamp. ("1677-09-22", datetime.date(1677, 9, 22)),