fix for error while loading state dict in cifar10 #1152

	# This workflow will install Python dependencies, run tests and lint with a single version of Python
	# For more information see: https://help.github.com/actions/language-and-framework-guides/using-python-with-github-actions

	name: TaskRunner (Eden Compression)

	on:
	pull_request:
	branches: [ develop ]

	permissions:
	contents: read

	jobs:
	build:
	if: contains(github.event.pull_request.labels.*.name, 'eden_compression')
	strategy:
	matrix:
	os: ['ubuntu-latest', 'windows-latest']
	runs-on: ${{ matrix.os }}

	steps:
	- uses: actions/checkout@v3
	- name: Set up Python 3.8
	uses: actions/setup-python@v3
	with:
	python-version: "3.8"
	- name: Install dependencies
	run: \|
	python -m pip install --upgrade pip
	pip install .
	- name: Test TaskRunner API with Eden Compression
	run: \|
	python -m tests.github.test_hello_federation --template torch_cnn_mnist_eden_compression --fed_workspace aggregator --col1 col1 --col2 col2 --rounds-to-train 3

Provide feedback