I don't think this fix works as intended, I don't see an issue with the click listener being lost on rotation. The only potential issue I see is the time being reset after rotation.

#4498

(Case 1) Catalog app:

1. Open a TimePicker.
   
3. Select any time.
   
4. Press OK button. -> The time is updated.
   

(Case 2) Catalog app:

1. Open a TimePicker.
2. Rotate the device.
3. Select any time.
   
4. Press OK button. -> The time is NOT updated.
   

https://github.com/material-components/material-components-android/blob/master/catalog/java/io/material/catalog/timepicker/TimePickerMainDemoFragment.java#L126-L133:

MaterialTimePicker materialTimePicker = materialTimePickerBuilder.build();
materialTimePicker.show(requireFragmentManager(), "fragment_tag");

materialTimePicker.addOnPositiveButtonClickListener(dialog -> {
  int newHour = materialTimePicker.getHour();
  int newMinute = materialTimePicker.getMinute();
  TimePickerMainDemoFragment.this.onTimeSet(newHour, newMinute);
});