Jorick,
In short, yes, you can use ChatGPT.
The purpose of homework is to make you familiar with data analysis process, and makes your hand dirty.
So students are allowed to use AI-based tools, such as ChatGPT.
But, as I mentioned before, students should take the response to check the suggested answer provided by AI, in order to make sure the answer (or a modified one) is exactly what you want.
And TA will grade your homework base on the analysis process you design and the explanation you give.
Thanks,
HSIAO