fix test

ludwig-ai · Aug 10, 2023 · 4ccacfa · 4ccacfa
1 parent 4d8f985
commit 4ccacfa
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/integration_tests/test_custom_components.py b/tests/integration_tests/test_custom_components.py
@@ -17,7 +17,7 @@
 from ludwig.modules.metric_modules import LossMetric, register_metric
 from ludwig.schema import utils as schema_utils
 from ludwig.schema.combiners.base import BaseCombinerConfig
-from ludwig.schema.combiners.utils import register_combiner as register_combiner_schema
+from ludwig.schema.combiners.utils import register_combiner_config
 from ludwig.schema.decoders.base import BaseDecoderConfig
 from ludwig.schema.decoders.utils import register_decoder_config
 from ludwig.schema.encoders.base import BaseEncoderConfig
@@ -55,7 +55,7 @@ class CustomLossConfig(BaseLossConfig):
     type: str = "custom_loss"
 
 
-@register_combiner_schema("custom_combiner")
+@register_combiner_config("custom_combiner")
 @dataclass
 class CustomTestCombinerConfig(BaseCombinerConfig):
     type: str = "custom_combiner"