Support older versions of Python. (External colab uses 3.7).

PiperOrigin-RevId: 482004157
google · Oct 18, 2022 · ca3f1ab · ca3f1ab
1 parent 0dbeb91
commit ca3f1ab
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/.github/workflows/build.yaml b/.github/workflows/build.yaml
@@ -14,7 +14,7 @@ jobs:
     runs-on: ubuntu-latest
     strategy:
       matrix:
-        python-version: [3.9]
+        python-version: [3.7, 3.8, 3.9]
     steps:
     - name: Cancel previous
       uses: styfle/[email protected]

diff --git a/orbax/checkpoint/future.py b/orbax/checkpoint/future.py
@@ -14,7 +14,8 @@
 
 """Orbax Future class used for duck typing."""
 
-from typing import Any, Optional, Protocol
+from typing import Any, Optional
+from typing_extensions import Protocol
 
 
 class Future(Protocol):