Changed model.names return type hint from list to dict[int, str] (#17402)

This commit is contained in:
Kilian Schnelle 2024-11-08 16:19:46 +01:00 committed by GitHub
parent 84bfe3c2e4
commit 95368cbe54
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -2,7 +2,7 @@
import inspect
from pathlib import Path
from typing import List, Union
from typing import Dict, List, Union
import numpy as np
import torch
@ -881,7 +881,7 @@ class Model(nn.Module):
return self
@property
def names(self) -> list:
def names(self) -> Dict[int, str]:
"""
Retrieves the class names associated with the loaded model.