Update exception in CreateModelRelease to properly indicate a populat…

…ion was not found.
world-federation-of-advertisers · Nov 13, 2024 · b18946d · b18946d
1 parent 01e8b2b
commit b18946d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/...kotlin/org/wfanet/measurement/kingdom/deploy/gcloud/spanner/writers/CreateModelRelease.kt b/...kotlin/org/wfanet/measurement/kingdom/deploy/gcloud/spanner/writers/CreateModelRelease.kt
@@ -29,6 +29,7 @@ import org.wfanet.measurement.gcloud.spanner.statement
 import org.wfanet.measurement.internal.kingdom.ModelRelease
 import org.wfanet.measurement.internal.kingdom.copy
 import org.wfanet.measurement.kingdom.deploy.gcloud.spanner.common.ModelSuiteNotFoundException
+import org.wfanet.measurement.kingdom.deploy.gcloud.spanner.common.PopulationNotFoundException
 
 class CreateModelRelease(private val modelRelease: ModelRelease) :
   SpannerWriter<ModelRelease, ModelRelease>() {
@@ -45,7 +46,7 @@ class CreateModelRelease(private val modelRelease: ModelRelease) :
     val externalPopulationId = ExternalId(modelRelease.externalPopulationId)
     val populationData: Struct =
       readPopulationData(externalDataProviderId, externalPopulationId)
-        ?: throw ModelSuiteNotFoundException(externalDataProviderId, externalPopulationId)
+        ?: throw PopulationNotFoundException(externalDataProviderId, externalPopulationId)
 
     val internalModelReleaseId = idGenerator.generateInternalId()
     val externalModelReleaseId = idGenerator.generateExternalId()