Skip to content

Commit a5af0d2

Browse files
committed
fix mypy and format
1 parent 55fefe7 commit a5af0d2

File tree

2 files changed

+4
-5
lines changed

2 files changed

+4
-5
lines changed

bigframes/core/array_value.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,6 @@
2121
import warnings
2222

2323
import google.cloud.bigquery
24-
import pandas
2524
import pyarrow as pa
2625

2726
import bigframes.core.expression as ex

tests/system/small/test_dataframe.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -3765,9 +3765,9 @@ def test_at_no_duplicate(scalars_df_index, scalars_pandas_df_index):
37653765

37663766

37673767
def test_setitem_w_timestamp_none():
3768-
b_df = bpd.DataFrame({'rowindex': [1, 2, 3]})
3769-
b_df['temp_timestamp'] = pd.Timestamp(ts_input=None, unit="us", tz="utc")
3770-
assert b_df['temp_timestamp'].dtype == "timestamp[us][pyarrow]"
3768+
b_df = bpd.DataFrame({"rowindex": [1, 2, 3]})
3769+
b_df["temp_timestamp"] = pd.Timestamp(ts_input=pd.NaT, unit="us", tz="utc") # type: ignore
3770+
assert b_df["temp_timestamp"].dtype == "timestamp[us][pyarrow]"
37713771

37723772

37733773
def test_loc_setitem_bool_series_scalar_new_col(scalars_dfs):
@@ -5720,4 +5720,4 @@ def test_agg_with_dict_containing_non_existing_col_raise_key_error(scalars_dfs):
57205720
}
57215721

57225722
with pytest.raises(KeyError):
5723-
bf_df.agg(agg_funcs)
5723+
bf_df.agg(agg_funcs)

0 commit comments

Comments
 (0)