Only test mode not working

gzerveas / mvts_transformer

Multivariate Time Series Transformer, public version

MIT License

727 stars 171 forks source link

Only test mode not working #1

Closed SylvainLan closed 2 years ago

SylvainLan commented 2 years ago

Hi,

I think there is a typo for the only test behaviour (--test_only testset), the input of pipeline factory should be pipeline_factory(config) instead of pipeline_factory(config['task']) which does not run

gzerveas commented 2 years ago

Thanks for spotting this! Merged the correction