add import of io

PiperOrigin-RevId: 757223579

Author: llcourage

lit_nlp/examples/gcp/model_server_test.py

@@ -13,6 +13,7 @@
 # limitations under the License.
 # ==============================================================================
 
+import io
 import os
 from unittest import mock
 
@@ -30,6 +31,7 @@ class TestWSGIApp(parameterized.TestCase):
   def setUpClass(cls):
     super().setUpClass()
 
+    _ = io.StringIO('temp data')
     test_model_name = 'lit_on_gcp_test_model'
     sal_name, tok_name = pd_utils.generate_model_group_names(test_model_name)
     test_model_config = f'{test_model_name}:test_model_path'

pyproject.toml

@@ -115,6 +115,8 @@ test = [
   "lime==0.2.0.1",
   "pytest>=7.4.0,<8.0.0",
   "webtest>=2.0",
+  "decorator>=4.4.0",
+  "contextlib2>=21.6.0",
 ]
 # LINT.ThenChange(./requirements_test.txt)
 

requirements_test.txt

@@ -20,4 +20,6 @@ lime==0.2.0.1
 pytest>=7.4.0,<8.0.0
 umap-learn==0.5.6
 webtest>=2.0
+decorator>=4.4.0
+contextlib2>=21.6.0
 # LINT.ThenChange(./pyproject.toml)