fixup: reverted to make linter pass

grpc · Aug 8, 2024 · 2d76f7c · 2d76f7c
1 parent 22d34a5
commit 2d76f7c
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/framework/helpers/docker.py b/framework/helpers/docker.py
@@ -285,9 +285,10 @@ def __init__(
         self.initial_resources = initial_resources
 
     def __enter__(self):
-        if super().__enter__():
-            self.update_resources(self.initial_resources)
-            return self
+        if not super().__enter__():
+            return None
+        self.update_resources(self.initial_resources)
+        return self
 
     def stop_on_resource_request(self, resource_type: str, resource_name: str):
         stub = xdsconfig_pb2_grpc.XdsConfigControlServiceStub(self.channel())