Add missing Optional on a few nullable fields.

This commit is contained in:
Alexandre Macabies
2023-07-30 16:25:12 +02:00
parent f17ad227cf
commit 50e00feceb
3 changed files with 6 additions and 5 deletions

View File

@ -276,7 +276,7 @@ class ModelInfo:
hash: str
location: Union[Path, str]
precision: torch.dtype
_cache: ModelCache = None
_cache: Optional[ModelCache] = None
def __enter__(self):
return self.context.__enter__()