diff options
| author | Adam Janovsky | 2023-04-14 13:37:32 +0200 |
|---|---|---|
| committer | Adam Janovsky | 2023-04-14 13:37:32 +0200 |
| commit | 17985924a176ab188e46f3ef4ca230f06dee23b5 (patch) | |
| tree | 9bc631e2f2757b6029fbbb0ee9b19297b65176c5 /src | |
| parent | 9e8198d1c0790ec69e309416a5b69559b340bca8 (diff) | |
| download | sec-certs-17985924a176ab188e46f3ef4ca230f06dee23b5.tar.gz sec-certs-17985924a176ab188e46f3ef4ca230f06dee23b5.tar.zst sec-certs-17985924a176ab188e46f3ef4ca230f06dee23b5.zip | |
fix cpe eval notebook
Diffstat (limited to 'src')
| -rw-r--r-- | src/sec_certs/dataset/dataset.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/sec_certs/dataset/dataset.py b/src/sec_certs/dataset/dataset.py index 8925e30b..4c530099 100644 --- a/src/sec_certs/dataset/dataset.py +++ b/src/sec_certs/dataset/dataset.py @@ -491,7 +491,7 @@ class Dataset(Generic[CertSubType, AuxiliaryDatasetsSubType], ComplexSerializabl cpe_dset = self._prepare_cpe_dataset() lst = [] - for cert in [x for x in cast(Iterator[Certificate], self) if x.heuristics.cpe_matches]: + for cert in [x for x in self if x.heuristics.cpe_matches]: dct = {"text": cert.label_studio_title} candidates = [cpe_dset[x].title for x in cert.heuristics.cpe_matches] candidates += ["No good match"] * (config.cpe_n_max_matches - len(candidates)) |