From b4683c665b4527c0958952b118a01ec8eda83429 Mon Sep 17 00:00:00 2001 From: mattahrens Date: Fri, 3 Nov 2023 14:14:17 -0500 Subject: [PATCH] Fixing Dataproc GKE costs for dataproc container Signed-off-by: mattahrens --- user_tools/src/spark_rapids_pytools/cloud_api/dataproc_gke.py | 2 +- .../src/spark_rapids_pytools/pricing/dataproc_gke_pricing.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/user_tools/src/spark_rapids_pytools/cloud_api/dataproc_gke.py b/user_tools/src/spark_rapids_pytools/cloud_api/dataproc_gke.py index 3243bd0e5..6039914bd 100644 --- a/user_tools/src/spark_rapids_pytools/cloud_api/dataproc_gke.py +++ b/user_tools/src/spark_rapids_pytools/cloud_api/dataproc_gke.py @@ -194,6 +194,6 @@ class DataprocGkeSavingsEstimator(DataprocSavingsEstimator): """ def _get_cost_per_cluster(self, cluster: ClusterGetAccessor): - dataproc_cost = super()._get_cost_per_cluster() + dataproc_cost = super()._get_cost_per_cluster(cluster) dataproc_gke_cost = self.price_provider.get_container_cost() return dataproc_cost + dataproc_gke_cost diff --git a/user_tools/src/spark_rapids_pytools/pricing/dataproc_gke_pricing.py b/user_tools/src/spark_rapids_pytools/pricing/dataproc_gke_pricing.py index 493da7e81..3c06e370b 100644 --- a/user_tools/src/spark_rapids_pytools/pricing/dataproc_gke_pricing.py +++ b/user_tools/src/spark_rapids_pytools/pricing/dataproc_gke_pricing.py @@ -28,7 +28,7 @@ class DataprocGkePriceProvider(DataprocPriceProvider): def get_container_cost(self) -> float: gke_container_cost = self.__get_gke_container_cost() - return dataproc_cost + gke_container_cost + return gke_container_cost def __get_gke_container_cost(self) -> float: lookup_key = 'CP-GKE-CONTAINER-MANAGMENT-COST'