[Train] Fix device_type attr bug (#1040) #11

	name: All Tests Metax

	on:
	push:
	branches: ["main"]
	paths:
	- '.github/workflows/metax.yml'
	- 'hardware/Metax_C550/**'
	- 'tests/**'
	pull_request:
	branches: ["main"]
	paths:
	- '.github/workflows/metax.yml'
	- 'hardware/Metax_C550/**'
	- 'tests/**'

	concurrency:
	group: ${{ github.workflow }}-${{ github.event.pull_request.number \|\| github.ref }}-${{ github.actor }}
	cancel-in-progress: true

	jobs:
	set-env:
	runs-on: ubuntu-latest
	outputs:
	ci_image: ${{ steps.set-env.outputs.ci_image }} # Declare output variable
	steps:
	- name: Set Environment Variable
	id: set-env # Assign an ID to this step
	run: \|
	echo "ci_image=localhost:5000/flagscale:maca.ai3.0.0.5-torch2.6-py310-ubuntu22.04-amd64-time2508201730" >> $GITHUB_OUTPUT # Set output variable

	functional-tests-inference:
	needs:
	- set-env
	uses: ./.github/workflows/functional-tests-metax.yml
	strategy:
	matrix:
	task:
	- deepseek_r1_distill_qwen-metax
	# TODO: Need update for [email protected]
	- deepseek_r1_distill_qwen-flaggems-metax
	- opi_llama3_1_instruct-metax
	- opi_llama3_1_instruct-flaggems-metax
	- qwen3-metax
	- qwen3-flaggems-metax
	- robobrain2-metax # TP1
	- robobrain2-flaggems-metax # TP1
	name: "inference-${{ matrix.task }}"
	with:
	task: ${{ matrix.task }}
	type: inference
	image: ${{ needs.set-env.outputs.ci_image }}

	# Check All Tests
	all-tests:
	needs:
	- functional-tests-inference
	runs-on: ubuntu-latest
	steps:
	- name: All Tests Completed
	run: echo "All tests completed successfully!"

Provide feedback