Change Clay forward signature to work with tasks #682

	name: terratorch tuning toolkit

	on:
	push:
	branches:
	- main

	pull_request:
	branches:
	- main

	jobs:
	build:
	runs-on: ubuntu-latest
	timeout-minutes: 20
	strategy:
	matrix:
	python-version: ["3.10", "3.11"]

	steps:
	- name: Clone repo
	uses: actions/checkout@v3
	- name: Set up Python ${{ matrix.python-version }}
	uses: actions/setup-python@v4
	with:
	python-version: ${{ matrix.python-version }}
	cache: 'pip'
	- name: Install dependencies
	run: \|
	python -m pip install --upgrade pip
	pip install -r requirements/required.txt -r requirements/test.txt -r requirements/optional.txt
	pip install git+https://github.com/NASA-IMPACT/Prithvi-WxC.git
	pip install git+https://github.com/IBM/granite-wxc.git
	- name: List pip dependencies
	run: pip list
	- name: Test with pytest
	run: \|
	pytest -s tests

	# build_weather:
	# runs-on: ubuntu-latest
	# strategy:
	# matrix:
	# python-version: ["3.11"]

	# steps:
	# - name: Clone repo
	# uses: actions/checkout@v3
	# - name: Set up Python ${{ matrix.python-version }}
	# uses: actions/setup-python@v4
	# with:
	# python-version: ${{ matrix.python-version }}
	# cache: 'pip'
	# - name: Install dependencies
	# run: \|
	# python -m pip install --upgrade pip
	# pip install -r requirements/required.txt -r requirements/test.txt -r requirements/optional.txt
	# pip install git+https://github.com/NASA-IMPACT/Prithvi-WxC.git
	# pip install git+https://github.com/IBM/granite-wxc.git
	# - name: List pip dependencies
	# run: pip list
	# - name: Test with pytest
	# run: \|
	# pytest -s tests

Provide feedback