Set distribution back.

project-koku · May 10, 2024 · fffba63 · fffba63
1 parent 1d08f9c
commit fffba63
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/koku/cost_models/serializers.py b/koku/cost_models/serializers.py
@@ -476,7 +476,11 @@ def validate(self, data):
             data["currency"] = get_currency(self.context.get("request"))
 
         if not data.get("distribution_info"):
-            data["distribution_info"] = metric_constants.DEFAULT_DISTRIBUTION_INFO
+            # TODO: Have this return just the default distribution info after
+            # QE updates tests.
+            distribution_info = metric_constants.DEFAULT_DISTRIBUTION_INFO
+            distribution_info["distribution_type"] = data.get("distribution", metric_constants.CPU_DISTRIBUTION)
+            data["distribution_info"] = distribution_info
         if (
             data.get("markup")
             and not data.get("rates")