Merge pull request #133 from MaRDI4NFDI/openml_tags

add tags as items instead of strings
MaRDI4NFDI · Mar 21, 2024 · c090252 · c090252
2 parents 9e2e975 + f4e0485
commit c090252
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/mardi_importer/mardi_importer/openml/OpenMLDataset.py b/mardi_importer/mardi_importer/openml/OpenMLDataset.py
@@ -162,7 +162,8 @@ def insert_claims(self):
                 prop_nr = self.api.get_local_id_by_label("OpenML semantic tag", "property")
                 tag_claims = []
                 for vt in valid_tags:
-                    claim = self.api.get_claim(prop_nr, vt)
+                    vt_id = self.api.get_local_id_by_label(vt, "item")
+                    claim = self.api.get_claim(prop_nr, vt_id)
                     tag_claims.append(claim)
                 self.item.add_claims(tag_claims)
         if self.paper_url and self.paper_url != "None":