Bug fix

Signed-off-by: Constantin M Adam <cmadam@us.ibm.com>
IBM · Oct 14, 2024 · c14bdaa · c14bdaa
1 parent 5e4022c
commit c14bdaa
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/transforms/universal/fdedup/spark/src/cluster_analysis_transform_spark.py b/transforms/universal/fdedup/spark/src/cluster_analysis_transform_spark.py
@@ -45,8 +45,8 @@ def get_folders(self, data_access: DataAccess) -> list[str]:
         :param data_access - data access object
         :return: list of folder paths
         """
-        bands = self.params["num_bands"]
-        segments = self.params["num_segments"]
+        bands = self.params[num_bands_key]
+        segments = self.params[num_segments_key]
         folders = [os.path.join(f"band={b}", f"segment={s}") for b in range(bands) for s in range(segments)]
         return folders