Move `logits.float()` call (#308) #983

	name: CI Pipeline

	on:
	push:
	branches:
	- main
	pull_request:
	branches:
	- main

	jobs:
	checkstyle:
	runs-on: ubuntu-latest

	steps:
	- name: Checkout code
	uses: actions/checkout@v3

	- name: Set up Python
	uses: actions/setup-python@v3
	with:
	python-version: '3.10'

	- name: Install dependencies
	run: \|
	python -m pip install --upgrade pip
	pip install flake8 isort black

	- name: Run checkstyle
	run: make checkstyle

Provide feedback