Update tests/pipelines/photon/test_pipeline_photon.py

DavidBert · dg845 · DavidBert · commit 5c54baacb7fd · 2025-10-21T07:29:30.000Z
Co-authored-by: dg845 &lt;58458699+dg845@users.noreply.github.com&gt;
diff --git a/tests/pipelines/photon/test_pipeline_photon.py b/tests/pipelines/photon/test_pipeline_photon.py
@@ -1,6 +1,7 @@
 import unittest
 
 import numpy as np
+import pytest
 import torch
 from transformers import AutoTokenizer
 from transformers.models.t5gemma.configuration_t5gemma import T5GemmaConfig, T5GemmaModuleConfig
