Address PR comments

Author: GianmarcoFolchi

Source/ASCollectionNode.mm

@@ -82,6 +82,7 @@ - (instancetype)init
     _flags.inverted = NO;
     _contentInset = UIEdgeInsetsZero;
     _contentOffset = CGPointZero;
+    _leadingScreensForBatching = 2.0;
     _flags.animatesContentOffset = NO;
     _flags.showsVerticalScrollIndicator = YES;
     _flags.showsHorizontalScrollIndicator = YES;
@@ -323,6 +324,7 @@ - (void)didLoad
     view.layoutInspector                = pendingState.layoutInspector;
     view.showsVerticalScrollIndicator   = pendingState.showsVerticalScrollIndicator;
     view.showsHorizontalScrollIndicator = pendingState.showsHorizontalScrollIndicator;
+    view.leadingScreensForBatching      = pendingState.leadingScreensForBatching;
 #if !TARGET_OS_TV
     view.pagingEnabled                  = pendingState.pagingEnabled;
 #endif
@@ -345,10 +347,6 @@ - (void)didLoad
       [view setContentOffset:contentOffset animated:pendingState.animatesContentOffset];
     }
 
-    if (pendingState.leadingScreensForBatching != 0) {
-      view.leadingScreensForBatching = pendingState.leadingScreensForBatching;
-    }
-    
     const auto tuningParametersVector = pendingState->_tuningParameters;
     const auto tuningParametersVectorSize = tuningParametersVector.size();
     for (NSInteger rangeMode = 0; rangeMode < tuningParametersVectorSize; rangeMode++) {

Source/ASTableNode.mm

@@ -163,6 +163,7 @@ - (void)didLoad
     view.allowsMultipleSelection              = pendingState.allowsMultipleSelection;
     view.allowsMultipleSelectionDuringEditing = pendingState.allowsMultipleSelectionDuringEditing;
     view.automaticallyAdjustsContentOffset    = pendingState.automaticallyAdjustsContentOffset;
+    view.leadingScreensForBatching            = pendingState.leadingScreensForBatching;
 #if !TARGET_OS_TV
     view.pagingEnabled                        = pendingState.pagingEnabled;
 #endif
@@ -176,10 +177,6 @@ - (void)didLoad
     if (!CGPointEqualToPoint(contentOffset, CGPointZero)) {
       [view setContentOffset:contentOffset animated:pendingState.animatesContentOffset];
     }
-
-     if (pendingState.leadingScreensForBatching != 0) {
-       view.leadingScreensForBatching = pendingState.leadingScreensForBatching;
-     }
 
     const auto tuningParametersVector = pendingState->_tuningParameters;
     const auto tuningParametersVectorSize = tuningParametersVector.size();