Move IAS logic into existing function

gso/workflows/l3_core_service/ias/create_imported_ias.py

@@ -19,7 +19,6 @@ from gso.workflows.l3_core_service.base_create_imported_l3_core_service import (
 from gso.workflows.l3_core_service.base_create_imported_l3_core_service import (
     initialize_subscription,
 )
-from gso.workflows.l3_core_service.ias.shared import update_ias_subscription_model
 
 
 def initial_input_form_generator() -> FormGenerator:
@@ -36,11 +35,14 @@ def initial_input_form_generator() -> FormGenerator:
 
 
 @step("Create subscription")
-def create_subscription(partner: str) -> dict:
+def create_subscription(partner: str, ias_flavor: IASFlavor) -> dict:
     """Create a new subscription object in the database."""
     partner_id = get_partner_by_name(partner).partner_id
     product_id = get_product_id_by_name(ProductName.IMPORTED_IAS)
     subscription = ImportedIASInactive.from_product_id(product_id, partner_id)
+
+    subscription.ias.ias_flavor = ias_flavor
+
     return {"subscription": subscription, "subscription_id": subscription.subscription_id}
 
 
@@ -54,7 +56,6 @@ def create_imported_ias() -> StepList:
     return (
         begin
         >> create_subscription
-        >> update_ias_subscription_model
         >> store_process_subscription(Target.CREATE)
         >> initialize_subscription
         >> set_status(SubscriptionLifecycle.ACTIVE)