BEGIN_PUBLIC

Get NimbusKit compiling with `-Woverriding-method-mismatch`
END_PUBLIC

PiperOrigin-RevId: 411635028

src/models/src/NICellCatalog.m

@@ -204,17 +204,23 @@ - (id)initWithStyle:(UITableViewCellStyle)style reuseIdentifier:(NSString *)reus
   return self;
 }
 
-- (BOOL)shouldUpdateCellWithObject:(NIDrawRectBlockCellObject *)object {
+- (BOOL)shouldUpdateCellWithObject:(id)object {
+  NSAssert([object isKindOfClass:[NIDrawRectBlockCellObject class]],
+           @"object must be NIDrawRectBlockCellObject");
+  NIDrawRectBlockCellObject *blockCell = (NIDrawRectBlockCellObject *)object;
   NIDrawRectBlockView* blockView = (NIDrawRectBlockView *)self.blockView;
-  blockView.block = object.block;
-  blockView.object = object.object;
+  blockView.block = blockCell.block;
+  blockView.object = blockCell.object;
   blockView.cell = self;
   [blockView setNeedsDisplay];
   return YES;
 }
 
-+ (CGFloat)heightForObject:(NIDrawRectBlockCellObject *)object atIndexPath:(NSIndexPath *)indexPath tableView:(UITableView *)tableView {
-  return object.block(tableView.bounds, object.object, nil);
++ (CGFloat)heightForObject:(id)object atIndexPath:(NSIndexPath *)indexPath tableView:(UITableView *)tableView {
+  NSAssert([object isKindOfClass:[NIDrawRectBlockCellObject class]],
+           @"object must be NIDrawRectBlockCellObject");
+  NIDrawRectBlockCellObject *blockCell = (NIDrawRectBlockCellObject *)object;
+  return blockCell.block(tableView.bounds, blockCell.object, nil);
 }
 
 @end