Skip to content

Commit 8cf668b

Browse files
committed
Update test
1 parent 8839c7a commit 8cf668b

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

test/core/test_worker_client_e2e.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,7 @@
88
from functools import wraps
99
from unittest.mock import patch, Mock
1010

11+
from rekcurd import RekcurdPack
1112
from rekcurd.utils import PredictResult
1213

1314

@@ -81,7 +82,7 @@ def assertArrStringResponse(self, response):
8182
def setUpClass(cls):
8283
server = grpc.server(futures.ThreadPoolExecutor(max_workers=1))
8384
rekcurd_pb2_grpc.add_RekcurdWorkerServicer_to_server(
84-
RekcurdWorkerServicer(app=app, predictor=None), server)
85+
RekcurdWorkerServicer(RekcurdPack(app, None)), server)
8586
server.add_insecure_port("[::]:5000")
8687
server.start()
8788
cls.server = server

0 commit comments

Comments
 (0)