code: def get_table_row_count(parquet_url: str):
      spark = get_spark_session()
      return spark.read.parquet(parquet_url).count()
  
imports: import logging
  from typing import Dict, List, Optional
  import pytorch_lightning
  import torch
  from petastorm import make_batch_reader
  from petastorm.pytorch import DataLoader, decimal_friendly_collate
  from reagent.core.oss_tensorboard_logger import OssTensorboardLogger
  from reagent.data.spark_utils import get_spark_session
  from reagent.preprocessing.batch_preprocessor import BatchPreprocessor
  from reagent.training import StoppingEpochCallback
  from reagent.training.reagent_lightning_module import has_test_step_override
  from .types import Dataset, ReaderOptions, ResourceOptions
label_desc: Use image node_id 1 for calling a global function with example usage: get_table_row_count(parquet_url) and returns: spark