Remove devcontainer

cvxgrp · Jul 25, 2023 · e72911e · e72911e
1 parent 325e135
commit e72911e
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 5 deletions.
diff --git a/cvx/bson/file.py b/cvx/bson/file.py
@@ -21,7 +21,7 @@ def build_bson(file):
     Returns:
         _BsonFile object.
     """
-    return _BsonFile(file, read_bson(file))
+    return _BsonFile(file, dict(read_bson(file)))
 
 
 @dataclass(frozen=True)

diff --git a/cvx/bson/io.py b/cvx/bson/io.py
@@ -28,9 +28,8 @@ def read_bson(file: Path):
         bson_document = openfile.read()
         data = bson.loads(bson_document)
 
-        return {
-            name: pd.read_feather(io.BytesIO(value)) for name, value in data.items()
-        }
+        for name, value in data.items():
+            yield name, pd.read_feather(io.BytesIO(value))
 
 
 def write_bson(file: Path, dic, compression="zstd"):

diff --git a/tests/test_bson.py b/tests/test_bson.py
@@ -12,7 +12,7 @@ def test_write(prices, tmp_path):
     write_bson(tmp_path / "maffay.bson", dic)
     assert (tmp_path / "maffay.bson").exists()
 
-    x = read_bson(tmp_path / "maffay.bson")
+    x = dict(read_bson(tmp_path / "maffay.bson"))
 
     assert set(x.keys()) == {"A", "B"}
     pd.testing.assert_frame_equal(x["A"], prices)