How can I close the message output during training and prediction?

[ZhikangLai]

Description

Link

No response

[jmoralez]

Add the following before calling fit:

import logging
import ray

logging.getLogger('pytorch_lightning').setLevel(logging.ERROR)
ray.init(log_to_driver=False)

For the id warning you have to set the env variable as the message says, but you have to handle the fact that the id will not be in the index anymore:

import os

os.environ['NIXTLA_ID_AS_COL'] = '1'
# don't reset the index on the output from predict if that's what you're currently doing

[ZhikangLai]

os.environ['NIXTLA_ID_AS_COL'] = '1'

Thank you. Mr. @jmoralez . It works!