Fix broken docs-build by pinning protobuf-3.20.x

Differential Revision: D72016216

Pull Request resolved: #1033

Author: kiukchung

.github/workflows/doc-build.yaml

@@ -20,16 +20,15 @@ jobs:
           architecture: x64
       - name: Checkout TorchX
         uses: actions/checkout@v2
-      - name: Install Dependencies
+      - name: Install Doc Dependencies
         run: |
           set -eux
           sudo apt-get update
           sudo apt-get install -y pandoc
-          pip install -e .[dev]
           pip install -r docs/requirements.txt
       - name: Install TorchX
         run: |
-          python setup.py develop
+          pip install -e .[dev]
       - name: Start Airflow
         run: |
           # start airflow in background

.github/workflows/pyre.yaml

@@ -22,6 +22,6 @@ jobs:
           set -eux
           pip install -e .[dev]
       - name: Init Lint Runner
-          lintrunner init
+        run: lintrunner init
       - name: Run Pyre
         run: scripts/pyre.sh

dev-requirements.txt

@@ -14,6 +14,8 @@ google-cloud-runtimeconfig==0.34.0
 hydra-core
 ipython
 kfp==1.8.22
+# pin protobuf to the version that is required by kfp
+protobuf==3.20.3
 mlflow-skinny
 moto~=5.0.8
 pyre-extensions

torchx/schedulers/test/ray_scheduler_test.py

@@ -390,7 +390,12 @@ def test_nonmatching_address(self) -> None:
             ):
                 _scheduler_with_client.submit(app=app, cfg={})
 
-        def _assertDictContainsSubset(self, expected, actual, msg=None):
+        def _assertDictContainsSubset(
+            self,
+            expected: dict[str, Any],
+            actual: dict[str, Any],
+            msg: Optional[str] = None,
+        ) -> None:
             # NB: implement unittest.TestCase.assertDictContainsSubsetNew() since it was removed in python-3.11
             for key, value in expected.items():
                 self.assertIn(key, actual, msg)