Missing exception import: PartialShardFailureException

Change-Id: Iff532ad17c8928142afbaa8b22aee1c1368c2b0d