add to the transformation steps

Author: codeloop

ads/opctl/operator/lowcode/anomaly/model/anomaly_dataset.py

@@ -4,16 +4,18 @@
 # Copyright (c) 2023 Oracle and/or its affiliates.
 # Licensed under the Universal Permissive License v 1.0 as shown at https://oss.oracle.com/licenses/upl/
 
-from ..operator_config import AnomalyOperatorSpec
+import pandas as pd
+
+from ads.opctl import logger
+from ads.opctl.operator.lowcode.anomaly.const import OutputColumns
+from ads.opctl.operator.lowcode.anomaly.utils import get_frequency_of_datetime
+from ads.opctl.operator.lowcode.common.data import AbstractData
 from ads.opctl.operator.lowcode.common.utils import (
     default_signer,
     merge_category_columns,
 )
-from ads.opctl.operator.lowcode.common.data import AbstractData
-from ads.opctl.operator.lowcode.anomaly.utils import get_frequency_of_datetime
-from ads.opctl import logger
-import pandas as pd
-from ads.opctl.operator.lowcode.anomaly.const import OutputColumns
+
+from ..operator_config import AnomalyOperatorSpec
 
 
 class AnomalyData(AbstractData):

ads/opctl/operator/lowcode/common/transformations.py

@@ -91,6 +91,8 @@ def run(self, data):
                     logger.info("Skipping outlier treatment because it is disabled")
             elif self.name == "additional_data":
                 clean_df = self._missing_value_imputation_add(clean_df)
+            elif self.name == "input_data" and self.preprocessing.steps.missing_value_imputation:
+                clean_df = self._fill_na(clean_df)
         else:
             logger.info(
                 "Skipping all preprocessing steps because preprocessing is disabled"