resolve unsoundness caused by pytype.

PiperOrigin-RevId: 842351622

Author: p3rf Team

tests/providers/aws/aws_dynamodb_test.py

@@ -81,7 +81,13 @@ def GetTestDynamoDBInstance(minimal=False):
       yaml_string=spec, benchmark_name='aws_dynamodb_ycsb'
   )
   test_benchmark_spec.ConstructNonRelationalDb()
-  return test_benchmark_spec.non_relational_db
+  instance = test_benchmark_spec.non_relational_db
+  if instance is None:
+    raise ValueError(
+        'BenchmarkSpec.ConstructNonRelationalDb() failed to create a '
+        'non_relational_db instance for testing, tests cannot proceed.'
+    )
+  return instance
 
 
 class AwsDynamodbTest(pkb_common_test_case.PkbCommonTestCase):

tests/providers/gcp/gcp_bigtable_test.py

@@ -107,10 +107,17 @@ def GetTestBigtableInstance(spec=_TEST_BENCHMARK_SPEC):
       yaml_string=spec, benchmark_name='cloud_bigtable_ycsb'
   )
   test_benchmark_spec.ConstructNonRelationalDb()
-  return test_benchmark_spec.non_relational_db
+  instance = test_benchmark_spec.non_relational_db
+  if instance is None:
+    raise ValueError(
+        'BenchmarkSpec.ConstructNonRelationalDb() failed to create a '
+        'non_relational_db instance for testing, tests cannot proceed.'
+    )
+  return instance
 
 
 class GcpBigtableTestCase(pkb_common_test_case.PkbCommonTestCase):
+  bigtable: gcp_bigtable.GcpBigtableInstance
 
   def setUp(self):
     super().setUp()

tests/scripts/messaging_service_scripts_e2e_workers_test.py

@@ -62,7 +62,7 @@ class BaseSubprocessTest(pkb_common_test_case.PkbCommonTestCase):
 
   @classmethod
   def Main(cls, *args, **kwargs):
-    cls.main_function(*args, **kwargs)  # pylint: disable=not-callable
+    cls.main_function(*args, **kwargs)  # pylint: disable=not-callable # pytype: disable=not-callable
 
   def setUp(self):
     super().setUp()