[Release] New v0.2.0 release (#163)

README.md

@@ -104,7 +104,7 @@ visual("cora", seed=0, depth=3)
 
 # load OAGBert model and perform inference
 oagbert = pipeline("oagbert")
-outputs = oagbert("CogDL is developed by KEG, Tsinghua.")
+outputs = oagbert(["CogDL is developed by KEG, Tsinghua.", "OAGBert is developed by KEG, Tsinghua."])
 ```
 
 ### Command-Line Usage

README_CN.md

@@ -100,7 +100,7 @@ visual("cora", seed=0, depth=3)
 
 # load OAGBert model and perform inference
 oagbert = pipeline("oagbert")
-outputs = oagbert("CogDL is developed by KEG, Tsinghua.")
+outputs = oagbert(["CogDL is developed by KEG, Tsinghua.", "OAGBert is developed by KEG, Tsinghua."])
 ```
 
 

cogdl/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "0.1.2"
+__version__ = "0.2.0"
 
 from .experiments import experiment
 from .oag import oagbert

cogdl/pipelines.py

@@ -121,7 +121,7 @@ def __init__(self, app: str, model: str, **kwargs):
         self.tokenizer, self.bert_model = oagbert(model, load_weights=load_weights)
 
     def __call__(self, sequence, **kwargs):
-        tokens = self.tokenizer(sequence, return_tensors="pt")
+        tokens = self.tokenizer(sequence, return_tensors="pt", padding=True)
         outputs = self.bert_model(**tokens)
 
         return outputs

examples/oagbert.py

@@ -2,8 +2,8 @@
 
 tokenizer, bert_model = oagbert()
 
-sequence = "CogDL is developed by KEG, Tsinghua."
-tokens = tokenizer(sequence, return_tensors="pt")
+sequence = ["CogDL is developed by KEG, Tsinghua.", "OAGBert is developed by KEG, Tsinghua."]
+tokens = tokenizer(sequence, return_tensors="pt", padding=True)
 outputs = bert_model(**tokens)
 
 print(outputs[0].shape)

examples/pipeline.py

@@ -10,4 +10,4 @@
 
 # load OAGBert model and perform inference
 oagbert = pipeline("oagbert")
-outputs = oagbert("CogDL is developed by KEG, Tsinghua.")
+outputs = oagbert(["CogDL is developed by KEG, Tsinghua.", "OAGBert is developed by KEG, Tsinghua."])