From 99a97195c4ff6f3f3c8daaf5f2af140ea215f646 Mon Sep 17 00:00:00 2001 From: Maxime Arthaud Date: Fri, 28 May 2021 03:08:34 -0700 Subject: [PATCH] mark fb303.fb303.FacebookService.Client as a context manager (#5542) --- stubs/fb303/fb303/FacebookService.pyi | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/stubs/fb303/fb303/FacebookService.pyi b/stubs/fb303/fb303/FacebookService.pyi index 73296cbc4..19a6a8ec5 100644 --- a/stubs/fb303/fb303/FacebookService.pyi +++ b/stubs/fb303/fb303/FacebookService.pyi @@ -1,4 +1,4 @@ -from typing import Any +from typing import Any, ContextManager from thrift.Thrift import TProcessor # type: ignore @@ -19,7 +19,7 @@ class Iface: def reinitialize(self): ... def shutdown(self): ... -class Client(Iface): +class Client(Iface, ContextManager[Client]): def __init__(self, iprot, oprot=...) -> None: ... def getName(self): ... def send_getName(self): ...