datarobot-user-models icon indicating copy to clipboard operation
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.

Open kijimoshib opened this issue 10 months ago • 3 comments

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:

  1. 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
  2. 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
  3. predict_mixin.py:

    • Added request.args as param to predict

kijimoshib avatar Feb 01 '25 01:02 kijimoshib