Flash Zero does not recognize arguments hidden in kwargs

[Borda]

🐛 Bug

if the from_csv has some inherited arguments such as batch_size it is not recognized...

To Reproduce

flash tabular_classification \
    --model.learning_rate=0.01 \
    --model.optimizer="AdamW" \
    --trainer.max_epochs 20 \
    --trainer.accumulate_grad_batches=12 \
    --trainer.gradient_clip_val=0.1 \
    from_csv \
    --train_file=/home/jirka/Downloads/train.csv \
    --numerical_fields=f0 \
    --target_fields="target" \
    --val_split=0.2 \
    --batch_size=512

fails with:

tabular_classification: error: Unrecognized arguments: --val_split=0.2 --batch_size=512

moreover seems as the kwargs are not exposed at all:

usage: tabular_classification [options] from_csv [-h] [--categorical_fields CATEGORICAL_FIELDS] [--numerical_fields NUMERICAL_FIELDS] [--target_fields TARGET_FIELDS]
                                                 [--parameters PARAMETERS] [--train_file TRAIN_FILE] [--val_file VAL_FILE] [--test_file TEST_FILE] [--predict_file PREDICT_FILE]
                                                 [--train_transform TRAIN_TRANSFORM] [--val_transform VAL_TRANSFORM] [--test_transform TEST_TRANSFORM]
                                                 [--predict_transform PREDICT_TRANSFORM]

Additional context