tp world_size fix #184

	# This workflow will install Python dependencies and run MyPy

	name: MyPy Type Checking

	on: [pull_request]

	permissions:
	contents: read

	jobs:
	build:

	runs-on: ubuntu-latest

	steps:
	- uses: actions/checkout@v3
	- name: Set up Python 3.10
	uses: actions/setup-python@v3
	with:
	python-version: "3.10"
	- name: Install dependencies
	run: \|
	python -m pip install --upgrade pip
	pip install -r test-requirements.txt

	- name: Test with pytest
	run: \|
	mypy --exclude hf --exclude testing fms_extras

Provide feedback