[Python] Fix input type for num_bytes (#25086)

* fix input type * use sys for size
apache · Jan 19, 2023 · 80d8d34 · 80d8d34
1 parent 60a9382
commit 80d8d34
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sdks/python/apache_beam/ml/inference/sklearn_inference.py b/sdks/python/apache_beam/ml/inference/sklearn_inference.py
@@ -150,7 +150,7 @@ def run_inference(
 
  return _convert_to_result(batch, predictions)
 
- def get_num_bytes(self, batch: Sequence[pandas.DataFrame]) -> int:
+ def get_num_bytes(self, batch: Sequence[numpy.ndarray]) -> int:
  """
  Returns:
  The number of bytes of data for a batch.