def test_initialize_state():
    """Tests that state is INITIALIZING after Trainer creation."""
    trainer = Trainer()
    assert trainer.state == TrainerState(status=TrainerStatus.INITIALIZING, fn=None, stage=None)
Tests that state is INITIALIZING after Trainer creation.
tests/tests_pytorch/trainer/test_states.py