mirror of
https://github.com/NicolasBohn/NexQuant.git
synced 2026-08-01 01:27:42 +00:00
XianBW
7733841f1c
* feat: Add spec_enabled configuration for data science settings * make spec alternative * change spec logic in exp_gen * remove some general texts * align --------- Co-authored-by: Young <afe.young@gmail.com> Co-authored-by: yuanteli <1957922024@qq.com>
34 lines
1.2 KiB
Python
34 lines
1.2 KiB
Python
from pydantic_settings import SettingsConfigDict
|
|
|
|
from rdagent.app.kaggle.conf import KaggleBasePropSetting
|
|
|
|
|
|
class DataScienceBasePropSetting(KaggleBasePropSetting):
|
|
model_config = SettingsConfigDict(env_prefix="DS_", protected_namespaces=())
|
|
|
|
# Main components
|
|
## Scen
|
|
scen: str = "rdagent.scenarios.data_science.scen.KaggleScen"
|
|
"""Scenario class for data mining model"""
|
|
|
|
## Workflow Related
|
|
consecutive_errors: int = 5
|
|
|
|
debug_timeout: int = 600
|
|
"""The timeout limit for running on debugging data"""
|
|
full_timeout: int = 3600
|
|
"""The timeout limit for running on full data"""
|
|
|
|
### specific feature
|
|
|
|
#### enable specification
|
|
spec_enabled: bool = True
|
|
# - [ ] rdagent/components/coder/data_science/raw_data_loader/__init__.py: make spec implementation optional
|
|
# - [ ] move spec responsibility into rdagent/scenarios/data_science/share.yaml
|
|
# - [ ] make all spec.md optional; but replace it with the test & responsibility. "spec/.*\.md".
|
|
# - [ ] replace yaml render with target test. "spec > .yaml data_science !out_spec !task_spec model_spec"
|
|
# - [ ] At the head of all tests, emphasis the function to be tested.
|
|
|
|
|
|
DS_RD_SETTING = DataScienceBasePropSetting()
|