Update protos

viamrobotics · Sep 24, 2024 · 1f0f5c5 · 1f0f5c5
1 parent 9854fae
commit 1f0f5c5
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tests/mocks/services.py b/tests/mocks/services.py
@@ -165,7 +165,6 @@
     UploadModuleFileResponse,
 )
 from viam.proto.app.billing import (
-    BillingServiceBase,
     GetCurrentMonthUsageRequest,
     GetCurrentMonthUsageResponse,
     GetInvoicePdfRequest,
@@ -174,6 +173,7 @@
     GetInvoicesSummaryResponse,
     GetOrgBillingInformationRequest,
     GetOrgBillingInformationResponse,
+    UnimplementedBillingServiceBase,
 )
 from viam.proto.app.data import (
     AddBinaryDataToDatasetByIDsRequest,
@@ -193,7 +193,6 @@
     BoundingBoxLabelsByFilterResponse,
     ConfigureDatabaseUserRequest,
     ConfigureDatabaseUserResponse,
-    DataServiceBase,
     DeleteBinaryDataByFilterRequest,
     DeleteBinaryDataByFilterResponse,
     DeleteBinaryDataByIDsRequest,
@@ -219,6 +218,7 @@
     TabularDataBySQLResponse,
     TagsByFilterRequest,
     TagsByFilterResponse,
+    UnimplementedDataServiceBase,
 )
 from viam.proto.app.dataset import (
     CreateDatasetRequest,
@@ -792,7 +792,7 @@ async def SetSmartMachineCredentials(
         await stream.send_message(SetSmartMachineCredentialsResponse())
 
 
-class MockData(DataServiceBase):
+class MockData(UnimplementedDataServiceBase):
     def __init__(
         self,
         tabular_response: List[DataClient.TabularData],
@@ -1120,7 +1120,7 @@ async def DeleteCompletedTrainingJob(
         await stream.send_message(DeleteCompletedTrainingJobResponse())
 
 
-class MockBilling(BillingServiceBase):
+class MockBilling(UnimplementedBillingServiceBase):
     def __init__(
         self,
         pdf: bytes,