update test script for protobuf 4 (fixes #698) (#704)

* update test script * pin pillow version for demo.py
lanpa · Jul 9, 2023 · 7693d1a · 7693d1a
1 parent 55ef907
commit 7693d1a
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 3 deletions.
diff --git a/.github/workflows/python-app.yml b/.github/workflows/python-app.yml
@@ -56,7 +56,7 @@ jobs:
         python-version: 3.9
     - name: Run examples
       run: |
-        pip install six tensorboard pytest matplotlib torchvision protobuf==4.22.3 moviepy==1.0.3 imageio==2.27
+        pip install six tensorboard pytest matplotlib torchvision protobuf==4.22.3 moviepy==1.0.3 imageio==2.27 pillow==9.5.0
         python examples/demo.py
         python examples/demo_graph.py
         python examples/demo_embedding.py

diff --git a/run_pytest.sh b/run_pytest.sh
@@ -13,4 +13,4 @@ if [ `ps -ef|grep visdom |wc -l` = "1" ]
     trap "kill -SIGTERM $!" EXIT
 fi
 
-PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION=python pytest
+pytest
diff --git a/setup.py b/setup.py
@@ -59,6 +59,7 @@ def run(self):
         'Programming Language :: Python :: 3.8',
         'Programming Language :: Python :: 3.9',
         'Programming Language :: Python :: 3.10',
+        'Programming Language :: Python :: 3.11',
     ],
     cmdclass={
         'develop': PostDevelopCommand,

diff --git a/test-requirements.txt b/test-requirements.txt
@@ -2,7 +2,7 @@ flake8
 pytest
 torch
 torchvision
-protobuf==3.20.3
+protobuf==4.22.3
 numpy
 tensorboard
 boto3
@@ -12,4 +12,5 @@ soundfile
 visdom
 onnx
 pytest-cov
+imageio==2.27