You need to record the row count as a metric named row_count that can be...

Log a numerical or string value to the run with the given name using log(name, value, description= ' ' ). Logging a metric to a run causes that metric to be stored in the run record in the experiment. You can log the same metric multiple times within a run, the result being considered a vector of that metric.

Example: run.log( " accuracy " , 0.95)