Fixing Dataproc GKE costs for dataproc container

Signed-off-by: mattahrens <[email protected]>
NVIDIA · Nov 3, 2023 · b4683c6 · b4683c6
1 parent a7592b4
commit b4683c6
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 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
@@ -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
@@ -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'