Skip to content

Commit fa37151

Browse files
committed
format check
1 parent d4d890c commit fa37151

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

awswrangler/neptune/_neptune.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -110,8 +110,10 @@ def execute_sparql(client: NeptuneClient, query: str) -> pd.DataFrame:
110110
df = None
111111
if "results" in data and "bindings" in data["results"]:
112112
df = pd.DataFrame(data["results"]["bindings"], columns=data.get("head", {}).get("vars"))
113+
113114
def _binding_value(d: Any) -> Any:
114115
return d.get("value") if isinstance(d, dict) else None
116+
115117
for col in df.columns:
116118
df[col] = df[col].apply(_binding_value)
117119
else:

tests/unit/test_neptune.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -475,9 +475,7 @@ def test_sparql_bindings_nan_safe() -> None:
475475
class _DummyClient:
476476
def read_sparql(self, _query: str) -> dict[str, Any]:
477477
return {
478-
"head": {
479-
"vars": ["book_id", "title", "checked_out", "last_updated", "category_code"]
480-
},
478+
"head": {"vars": ["book_id", "title", "checked_out", "last_updated", "category_code"]},
481479
"results": {
482480
"bindings": [
483481
{

0 commit comments

Comments
 (0)