fmt

cvxgrp · Jul 27, 2023 · 3d1c2a3 · 3d1c2a3
1 parent fc4e615
commit 3d1c2a3
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/cvx/bson/file.py b/cvx/bson/file.py
@@ -79,5 +79,4 @@ def from_bson(bson_str: bytes) -> MATRICES:
     data = bson.loads(bson_str)
 
     # for name, value in data.items():
-    return {name: pa.ipc.read_tensor(value).to_numpy() for name, value in
-            data.items()}
+    return {name: pa.ipc.read_tensor(value).to_numpy() for name, value in data.items()}
diff --git a/tests/test_json.py b/tests/test_json.py
@@ -5,7 +5,9 @@
 from cvx.json import read_json, write_json
 
 
-@pytest.mark.parametrize("shape", [(50, 50), (1000, 50), (50, 1000), (1000, 1000), (5000, 2000)])
+@pytest.mark.parametrize(
+    "shape", [(50, 50), (1000, 50), (50, 1000), (1000, 1000), (5000, 2000)]
+)
 def test_read_and_write_json(tmp_path, shape):
     data = {"a": np.random.rand(*shape), "b": 3.0, "c": "test"}
     write_json(tmp_path / "test.json", data)