Minimal fix to fix type checking for latest pycose (#5869)

microsoft · Dec 18, 2023 · 6a7b6f2 · 6a7b6f2
1 parent 7b3f22f
commit 6a7b6f2
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/python/ccf/cose.py b/python/ccf/cose.py
@@ -4,15 +4,15 @@
 import argparse
 import sys
 
-from typing import Optional
+from typing import Optional, Type
 
 import base64
-import cbor2  # type: ignore
+import cbor2
 import json
 from datetime import datetime
 import pycose.headers  # type: ignore
 from pycose.keys.ec2 import EC2Key  # type: ignore
-from pycose.keys.curves import P256, P384, P521  # type: ignore
+from pycose.keys.curves import P256, P384, P521, CoseCurve  # type: ignore
 from pycose.keys.keyparam import EC2KpCurve, EC2KpX, EC2KpY, EC2KpD  # type: ignore
 from pycose.messages import Sign1Message  # type: ignore
 from cryptography.hazmat.primitives.asymmetric import ec
@@ -51,6 +51,7 @@ def from_cryptography_eckey_obj(ext_key) -> EC2Key:
         priv_nums = None
         pub_nums = ext_key.public_numbers()
 
+    curve: Type[CoseCurve]
     if pub_nums.curve.name == "secp256r1":
         curve = P256
     elif pub_nums.curve.name == "secp384r1":