Fix SIM117

OpenMined · Jun 29, 2024 · 0abb7c1 · 0abb7c1
1 parent 5c3a3ab
commit 0abb7c1
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/packages/syft/src/syft/serde/capnp.py b/packages/syft/src/syft/serde/capnp.py
@@ -10,6 +10,8 @@
 
 
 def get_capnp_schema(schema_file: str) -> type:
-    with as_file(files("syft.capnp").joinpath(schema_file)) as capnp_path:
-        with std_stream_capture():
-            return capnp.load(str(capnp_path.absolute()))
+    with (
+        as_file(files("syft.capnp").joinpath(schema_file)) as capnp_path,
+        std_stream_capture(),
+    ):
+        return capnp.load(str(capnp_path.absolute()))