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

googleapis · Apr 29, 2022 · 1ce2e54 · 1ce2e54
1 parent 64802e1
commit 1ce2e54
Showing 1 changed file with 1 addition and 0 deletions.
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)