Pull request #8: eda 모듈 변경으로 인한 mlps 수정

Merge in SLAISM/autoape-mlps from feature/SLCAI-221-autoape-eda to develop * commit 'a93a7815e124103f6b781d70e636b01c40c12ff3': eda 모듈 변경으로 인한 mlps 수정
seculayer · Sep 5, 2023 · 2d281dc · 2d281dc
2 parents d5d2eba + a93a781
commit 2d281dc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mlps/core/data/dataloader/DataLoaderAbstract.py b/mlps/core/data/dataloader/DataLoaderAbstract.py
@@ -102,7 +102,7 @@ def get_output_units(self, fields: List[FieldInfo]):
             self.LOGGER.info(field_info.stat_dict)
             if field_info.is_label:
                 try:
-                    return len(field_info.stat_dict.get("unique", {}))
+                    return field_info.stat_dict.get("unique").get("unique_count")
                 except Exception as e:
                     self.LOGGER.error(e, exc_info=True)
                     return 1