Fix TorchFTCheckpointManager's contract with BaseCheckpointManager (#4183)