mirror of
https://github.com/wassname/pytorch-ts.git
synced 2026-07-18 12:40:51 +08:00
fix source context
This commit is contained in:
@@ -84,7 +84,7 @@ class FileDataset(Dataset):
|
||||
for path in self.files():
|
||||
for line in JsonLinesFile(path):
|
||||
data = self.process(line.content)
|
||||
data["source"] = SourceContext(source=line.span, row=line.span.line)
|
||||
data["source"] = SourceContext(source=line.span.path, row=line.span.line)
|
||||
yield data
|
||||
|
||||
def __len__(self):
|
||||
|
||||
Reference in New Issue
Block a user