✅ Fix tgis client fixture for acceting kwargs

Signed-off-by: gkumbhat <[email protected]>
flaviabeo · May 17, 2024 · 48216eb · 48216eb
1 parent 40fb403
commit 48216eb
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/fixtures/__init__.py b/tests/fixtures/__init__.py
@@ -198,13 +198,13 @@ class StubTGISClient:
  def __init__(self, base_model_name):
  pass
 
- def Generate(self, request):
+ def Generate(self, request, **kwargs):
  return StubTGISClient.unary_generate(request)
 
- def GenerateStream(self, request):
+ def GenerateStream(self, request, **kwargs):
  return StubTGISClient.stream_generate(request)
 
- def Tokenize(self, request):
+ def Tokenize(self, request, **kwargs):
  return StubTGISClient.tokenize(request)
 
  @staticmethod