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

@@ -41,6 +41,10 @@ dependencies = [
   "termcolor>=2.3.0",
   "tqdm>=4.64.0",
   "werkzeug>=2.2.3",
+  "decorator>=4.4.0",
+  "contextlib2>=21.6.0",
+  "fsspec>=2023.9.0",
+  "colorama>=0.4.4",  
 ]
 # LINT.ThenChange(./requirements.txt)
 classifiers = [

requirements.txt

@@ -38,4 +38,8 @@ six>=1.16.0
 termcolor>=2.3.0
 tqdm>=4.64.0
 werkzeug>=2.2.3
+decorator>=4.4.0,
+contextlib2>=21.6.0,
+fsspec>=2023.9.0,
+colorama>=0.4.4,  
 # LINT.ThenChange(./pyproject.toml)