Fix FileNotFoundError

✅ Solution

FileNotFoundError in PyTorch tests often arises when the test attempts to load a file at a path that doesn't exist in the test environment or has been moved/deleted. Ensure the required file exists at the specified path, and verify the test's working directory or filepaths are correctly configured for the testing environment (e.g. adjust relative paths or environment variables pointing to the file). If the file is generated during the test, confirm the generation step occurred successfully before attempting to load it.