Cleanups

Author: rhshadrach

pandas/io/pytables.py

@@ -87,6 +87,7 @@
     PeriodArray,
 )
 from pandas.core.arrays.datetimes import tz_to_dtype
+from pandas.core.arrays.string_ import BaseStringArray
 import pandas.core.common as com
 from pandas.core.computation.pytables import (
     PyTablesExpr,
@@ -3226,8 +3227,6 @@ def write_array(
                 # get the atom for this datatype
                 atom = _tables().Atom.from_dtype(value.dtype)
 
-        from pandas.core.arrays.string_ import BaseStringArray
-
         if atom is not None:
             # We only get here if self._filters is non-None and
             #  the Atom.from_dtype call succeeded

pandas/tests/io/pytables/test_put.py

@@ -195,7 +195,7 @@ def test_put_compression_blosc(setup_path):
 
 
 def test_put_datetime_ser(setup_path, performance_warning, using_infer_string):
-    # https://github.com/pandas-dev/pandas/pull/???
+    # https://github.com/pandas-dev/pandas/pull/60663
     ser = Series(3 * [Timestamp("20010102").as_unit("ns")])
     with ensure_clean_store(setup_path) as store:
         store.put("ser", ser)
@@ -236,6 +236,7 @@ def test_put_mixed_type(setup_path, performance_warning, using_infer_string):
 
 
 def test_put_str_frame(setup_path, performance_warning, string_dtype_arguments):
+    # https://github.com/pandas-dev/pandas/pull/60663
     dtype = pd.StringDtype(*string_dtype_arguments)
     df = DataFrame({"a": pd.array(["x", pd.NA, "y"], dtype=dtype)})
     with ensure_clean_store(setup_path) as store:
@@ -248,6 +249,7 @@ def test_put_str_frame(setup_path, performance_warning, string_dtype_arguments):
 
 
 def test_put_str_series(setup_path, performance_warning, string_dtype_arguments):
+    # https://github.com/pandas-dev/pandas/pull/60663
     dtype = pd.StringDtype(*string_dtype_arguments)
     ser = Series(["x", pd.NA, "y"], dtype=dtype)
     with ensure_clean_store(setup_path) as store: