Correct function name in package import #606

	name: Test PyTorch 2.2 and 2.6

	on:
	push:
	branches: [ main ]
	paths:
	- '*/.py' # Run workflow for changes in Python files
	- '*/.ipynb'
	- '*/.yaml'
	- '*/.yml'
	- '*/.sh'
	pull_request:
	branches: [ main ]
	paths:
	- '*/.py'
	- '*/.ipynb'
	- '*/.yaml'
	- '*/.yml'
	- '*/.sh'

	jobs:
	test:
	runs-on: ubuntu-latest
	strategy:
	matrix:
	pytorch-version: [ 2.3.0, 2.6.0 ]

	steps:
	- uses: actions/checkout@v4

	- name: Set up Python
	uses: actions/setup-python@v5
	with:
	python-version: "3.13"

	- name: Install dependencies
	run: \|
	curl -LsSf https://astral.sh/uv/install.sh \| sh
	uv sync --dev --python=3.10 # tests for backwards compatibility
	uv pip install -r ch05/07_gpt_to_llama/tests/test-requirements-extra.txt
	uv add torch==${{ matrix.pytorch-version }}
	uv add pytest-ruff nbval

	- name: Test Selected Python Scripts
	run: \|
	source .venv/bin/activate
	pytest --ruff setup/02_installing-python-libraries/tests.py
	pytest --ruff ch04/01_main-chapter-code/tests.py
	pytest --ruff ch05/01_main-chapter-code/tests.py
	pytest --ruff ch05/07_gpt_to_llama/tests/tests.py
	pytest --ruff ch06/01_main-chapter-code/tests.py

	- name: Validate Selected Jupyter Notebooks
	run: \|
	source .venv/bin/activate
	pytest --ruff --nbval ch02/01_main-chapter-code/dataloader.ipynb
	pytest --ruff --nbval ch03/01_main-chapter-code/multihead-attention.ipynb
	pytest --ruff --nbval ch02/04_bonus_dataloader-intuition/dataloader-intuition.ipynb

Provide feedback