<｜begin▁of▁sentence｜>### Instruction:
Given an array input_data, how do you ensure that it's a square matrix before further processing in a graspologic pipeline?
### Response:
import numpy as np
from graspologic.preconditions import check_argument

# Check if the input data is a square matrix
input_data = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
square = (input_data.shape[0] == input_data.shape[1])

# Use the check_argument to ensure this is a square matrix
check_argument(square, 'Input data must be a square matrix.')
<|EOT|>
