Modify LLM Trainer to support BERT and Tiny LLaMA (#2031) #98

	name: Python Test

	on:
	- push
	- pull_request

	jobs:
	python-test:
	name: Test
	runs-on: ubuntu-latest

	steps:
	- name: Check out code
	uses: actions/checkout@v4

	# TODO (andreyvelich): We need to replace this action with script to do
	# linting and formatting for Training Operator SDK.
	- name: Check Python code with Black
	uses: psf/black@stable
	with:
	version: 24.2.0
	options: --check --exclude '/kubeflow_org_v1\|__init__.py\|api_client.py\|configuration.py\|exceptions.py\|rest.py'
	src: sdk/

	- name: Install dependencies
	run: \|
	pip install pytest python-dateutil urllib3 kubernetes
	pip install -U './sdk/python[huggingface]'

	- name: Run unit test for training sdk
	run: pytest ./sdk/python/kubeflow/training/api/training_client_test.py

Provide feedback