small fixes

Author: Jack Dermody

BrightData/Buffer/Composite/CompositeBufferBase.cs

@@ -176,6 +176,7 @@ public override async Task<ReadOnlyMemory<T>> GetTypedBlock(uint blockIndex)
                         byteOffset += await SkipFileBlock(_currentDataBlock, byteOffset, _fileBlockSizes[fileBlockIndex++]);
                     }
                 }
+                currentIndex = inMemoryBlocks + inFileBlocks;
             }
 
             // check current block

BrightData/LinearAlgebra/MutableMatrix.cs

@@ -80,31 +80,31 @@ protected set
         /// <inheritdoc />
         public INumericSegment<T> GetRow(uint index)
         {
-            if (index > RowCount)
+            if (index >= RowCount)
                 throw new ArgumentOutOfRangeException(nameof(index), $"Number of rows is {RowCount} but index {index} was requested");
             return new MutableTensorSegmentWrapper<T>(Segment, index, RowCount, ColumnCount);
         }
 
         /// <inheritdoc />
         public virtual INumericSegment<T> GetColumn(uint index)
         {
-            if (index > ColumnCount)
+            if (index >= ColumnCount)
                 throw new ArgumentOutOfRangeException(nameof(index), $"Number of columns is {ColumnCount} but index {index} was requested");
             return new MutableTensorSegmentWrapper<T>(Segment, index * RowCount, 1, RowCount);
         }
 
         /// <inheritdoc />
         public virtual IReadOnlyNumericSegment<T> GetReadOnlyRow(uint index)
         {
-            if (index > RowCount)
+            if (index >= RowCount)
                 throw new ArgumentOutOfRangeException(nameof(index), $"Number of rows is {RowCount} but index {index} was requested");
             return new ReadOnlyTensorSegmentWrapper<T>(Segment, index, RowCount, ColumnCount);
         }
 
         /// <inheritdoc />
         public virtual IReadOnlyNumericSegment<T> GetReadOnlyColumn(uint index)
         {
-            if (index > ColumnCount)
+            if (index >= ColumnCount)
                 throw new ArgumentOutOfRangeException(nameof(index), $"Number of columns is {ColumnCount} but index {index} was requested");
             return new ReadOnlyTensorSegmentWrapper<T>(Segment, index * RowCount, 1, RowCount);
         }