Skip to content

GitLab

Unverified Commit faa1dc56 authored by xiang song(charlie.song)'s avatar xiang song(charlie.song) Committed by GitHub
Browse files

hot fix (#1888) 


Co-authored-by: default avatarUbuntu <ubuntu@ip-172-31-51-214.ec2.internal>
parent aee10679
Show whitespace changes
Inline Side-by-side
Showing with 7 additions and 7 deletions
python/dgl/data/rdf.py
View file @ faa1dc56
...@@ -610,13 +610,13 @@ class AIFBDataset(RDFGraphDataset): ...@@ -610,13 +610,13 @@ class AIFBDataset(RDFGraphDataset):
def parse_entity(self, term): def parse_entity(self, term):
if isinstance(term, rdf.Literal): if isinstance(term, rdf.Literal):
return Entity(id=str(term), cls="_Literal") return Entity(e_id=str(term), cls="_Literal")
if isinstance(term, rdf.BNode): if isinstance(term, rdf.BNode):
return None return None
entstr = str(term) entstr = str(term)
if entstr.startswith(self.entity_prefix): if entstr.startswith(self.entity_prefix):
sp = entstr.split('/') sp = entstr.split('/')
return Entity(id=sp[5], cls=sp[3]) return Entity(e_id=sp[5], cls=sp[3])
else: else:
return None return None
...@@ -732,7 +732,7 @@ class MUTAGDataset(RDFGraphDataset): ...@@ -732,7 +732,7 @@ class MUTAGDataset(RDFGraphDataset):
def parse_entity(self, term): def parse_entity(self, term):
if isinstance(term, rdf.Literal): if isinstance(term, rdf.Literal):
return Entity(id=str(term), cls="_Literal") return Entity(e_id=str(term), cls="_Literal")
elif isinstance(term, rdf.BNode): elif isinstance(term, rdf.BNode):
return None return None
entstr = str(term) entstr = str(term)
...@@ -744,7 +744,7 @@ class MUTAGDataset(RDFGraphDataset): ...@@ -744,7 +744,7 @@ class MUTAGDataset(RDFGraphDataset):
cls = 'bond' cls = 'bond'
else: else:
cls = None cls = None
return Entity(id=inst, cls=cls) return Entity(e_id=inst, cls=cls)
else: else:
return None return None
...@@ -881,7 +881,7 @@ class BGSDataset(RDFGraphDataset): ...@@ -881,7 +881,7 @@ class BGSDataset(RDFGraphDataset):
# instance # instance
cls = '%s/%s' % (sp[4], sp[5]) cls = '%s/%s' % (sp[4], sp[5])
inst = sp[6] inst = sp[6]
return Entity(id=inst, cls=cls) return Entity(e_id=inst, cls=cls)
else: else:
return None return None
...@@ -1006,7 +1006,7 @@ class AMDataset(RDFGraphDataset): ...@@ -1006,7 +1006,7 @@ class AMDataset(RDFGraphDataset):
if isinstance(term, rdf.Literal): if isinstance(term, rdf.Literal):
return None return None
elif isinstance(term, rdf.BNode): elif isinstance(term, rdf.BNode):
return Entity(id=str(term), cls='_BNode') return Entity(e_id=str(term), cls='_BNode')
entstr = str(term) entstr = str(term)
if entstr.startswith(self.entity_prefix): if entstr.startswith(self.entity_prefix):
sp = entstr.split('/') sp = entstr.split('/')
...@@ -1018,7 +1018,7 @@ class AMDataset(RDFGraphDataset): ...@@ -1018,7 +1018,7 @@ class AMDataset(RDFGraphDataset):
else: else:
cls = 'TYPE' cls = 'TYPE'
inst = spp inst = spp
return Entity(id=inst, cls=cls) return Entity(e_id=inst, cls=cls)
else: else:
return None return None
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment