fix: test cases

Signed-off-by: Nandita Koppisetty <[email protected]>

tests/udfs/test_pipeline.py

@@ -8,7 +8,7 @@
 
 from numalogic._constants import TESTS_DIR
 from numalogic.udfs._config import PipelineConf
-from numalogic.udfs.pipeline import PipelineUDF
+from numalogic.udfs.ml_pipeline import MLPipelineUDF
 from tests.udfs.utility import input_json_from_file
 
 logging.basicConfig(level=logging.DEBUG)
@@ -31,8 +31,8 @@ def setUp(self) -> None:
         schema = OmegaConf.structured(PipelineConf)
         pl_conf = PipelineConf(**OmegaConf.merge(schema, _given_conf))
         pl_conf_2 = PipelineConf(**OmegaConf.merge(schema, _given_conf_2))
-        self.udf1 = PipelineUDF(pl_conf=pl_conf)
-        self.udf2 = PipelineUDF(pl_conf=pl_conf_2)
+        self.udf1 = MLPipelineUDF(pl_conf=pl_conf)
+        self.udf2 = MLPipelineUDF(pl_conf=pl_conf_2)
         self.udf1.register_conf("druid-config", pl_conf.stream_confs["druid-config"])
         self.udf2.register_conf("druid-config", pl_conf_2.stream_confs["druid-config"])