test: fix KeyError in test_rest_streaming.py (#373)

parthea · web-flow · commit 1ce2e5409e5b · 2022-04-29T10:32:45.000-07:00
diff --git a/tests/unit/test_rest_streaming.py b/tests/unit/test_rest_streaming.py
@@ -28,6 +28,7 @@
 from google.protobuf import timestamp_pb2
 
 
+__protobuf__ = proto.module(package=__name__)
 SEED = int(time.time())
 logging.info(f"Starting rest streaming tests with random seed: {SEED}")
 random.seed(SEED)
