Clean up tavernhook

taverntesting · Jan 18, 2024 · da53616 · da53616
1 parent 48b684a
commit da53616
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tavern/_plugins/grpc/tavernhook.py b/tavern/_plugins/grpc/tavernhook.py
@@ -4,6 +4,7 @@
 import yaml
 
 from tavern._core.dict_util import format_keys
+from tavern._core.pytest.config import TestConfig
 
 from .client import GRPCClient
 from .request import GRPCRequest
@@ -18,8 +19,7 @@
 request_block_name = "grpc_request"
 
 
-def get_expected_from_request(response_block, test_block_config, session):
-    # format so we can subscribe to the right topic
+def get_expected_from_request(response_block, test_block_config: TestConfig, session):
     f_expected = format_keys(response_block, test_block_config.variables)
     expected = f_expected