code: def check_partition_column(partition_column, cols):
      
      for k, v in cols.items():
          if k == partition_column:
              if v == "int":
                  return
              else:
                  raise InvalidPartitionColumn(
                      "partition_column must be int, and not {0}".format(v)
                  )
      raise InvalidPartitionColumn(
          "partition_column {0} not found in the query".format(partition_column)
      )
docs: Check partition_column existence and type
  
      Args:
          partition_column: partition_column name
          cols: dict with columns names and python types
  
      Returns:
          None