diff --git a/pipelines/config.py b/pipelines/config.py index 3bfd32c..dbb34c0 100644 --- a/pipelines/config.py +++ b/pipelines/config.py @@ -51,6 +51,7 @@ class FinetuneConfig: ) +@dataclass class BenchmarkConfig: """Top-level benchmark configuration loaded from TOML.""" @@ -102,6 +103,7 @@ class OpenAIConfig: ) +@dataclass class BertTopicTrainConfig: """BERTopic training configuration loaded from TOML."""