datarobot-user-models
datarobot-user-models copied to clipboard
[AGENT-6608][AGENT-6612][AGENT-6613] Add use_datarobot_predict and time series args to server and score command parsers.
This repository is public. Do not put here any private DataRobot or customer's data: code, datasets, model artifacts, .etc.
Rationale
The changes ensure that time series parameters:
- Only work with the score command
- Require --use-datarobot-predict flag
- Support proper ISO-8601 timestamp validation
- Have proper mutually exclusive options validation
- Are properly passed through the entire pipeline
Summary
Here's a summary of the main changes to implement time series parameters:
-
args_parser.py:
- Added time series validation to only allow in score command
- Modified _reg_arg_time_series to check command type
- Updated register_args to only attach time series args to score_parser
- Added command validation in verify_options
-
drum_score_adapter.py:
- Added new time series parameters:
- forecast_point
- predictions_start_date
- predictions_end_date
- Added documentation for new parameters
- Added instance variables for time series parameters
- Added new time series parameters:
-
predict_mixin.py:
- Added
request.argsas param to predict
- Added