Skip to content

Commit 5b8f379

Browse files
1 parent 5df28eb commit 5b8f379

File tree

2 files changed

+6
-10
lines changed

2 files changed

+6
-10
lines changed

Diff for: db_dtypes/json.py

+6-9
Original file line numberDiff line numberDiff line change
@@ -34,6 +34,7 @@
3434
"ge": pyarrow.compute.greater_equal,
3535
}
3636

37+
3738
@pd.api.extensions.register_extension_dtype
3839
class JSONDtype(pd.api.extensions.ExtensionDtype):
3940
"""Extension dtype for BigQuery JSON data."""
@@ -255,28 +256,24 @@ def _reduce(
255256
raise TypeError("JSONArray does not support min/max reducntion.")
256257
super()._reduce(name, skipna=skipna, keepdims=keepdims, **kwargs)
257258

258-
def __array__(
259-
self, dtype = None, copy = None
260-
) -> np.ndarray:
259+
def __array__(self, dtype=None, copy=None) -> np.ndarray:
261260
"""Correctly construct numpy arrays when passed to `np.asarray()`."""
262261
return self.to_numpy(dtype=dtype)
263262

264-
def to_numpy(self, dtype = None, copy = False, na_value = pd.NA) -> np.ndarray:
263+
def to_numpy(self, dtype=None, copy=False, na_value=pd.NA) -> np.ndarray:
265264
dtype, na_value = self._to_numpy_dtype_inference(dtype, na_value, self._hasna)
266265
pa_type = self._pa_array.type
267266
if not self._hasna or pd.isna(na_value) or pa.types.is_null(pa_type):
268267
data = self
269268
else:
270269
data = self.fillna(na_value)
271270
result = np.array(list(data), dtype=dtype)
272-
271+
273272
if data._hasna:
274273
result[data.isna()] = na_value
275274
return result
276275

277-
def _to_numpy_dtype_inference(
278-
self, dtype, na_value, hasna
279-
):
276+
def _to_numpy_dtype_inference(self, dtype, na_value, hasna):
280277
if dtype is not None:
281278
dtype = np.dtype(dtype)
282279

@@ -290,4 +287,4 @@ def _to_numpy_dtype_inference(
290287
na_value = np.timedelta64("nat")
291288
else:
292289
na_value = self.dtype.na_value
293-
return dtype, na_value
290+
return dtype, na_value

Diff for: tests/compliance/json/test_json_compliance.py

-1
Original file line numberDiff line numberDiff line change
@@ -251,7 +251,6 @@ def test_setitem_mask_boolean_array_with_na(self, data, box_in_series):
251251
super().test_setitem_mask_boolean_array_with_na(data, box_in_series)
252252

253253
@pytest.mark.parametrize("setter", ["loc", "iloc"])
254-
255254
@pytest.mark.xfail(reason="TODO: open an issue for ArrowExtentionArray")
256255
def test_setitem_scalar(self, data, setter):
257256
super().test_setitem_scalar(data, setter)

0 commit comments

Comments
 (0)