diff --git a/src/hipool/chunk.py b/src/hipool/chunk.py index 22ae70b..e3f7add 100644 --- a/src/hipool/chunk.py +++ b/src/hipool/chunk.py @@ -4,7 +4,6 @@ k: number of chunks in document c: chunk length (in tokens) - """ import torch diff --git a/src/hipool/curiam_reader.py b/src/hipool/curiam_reader.py index 823a5b3..88aea27 100644 --- a/src/hipool/curiam_reader.py +++ b/src/hipool/curiam_reader.py @@ -5,7 +5,6 @@ The labels for each document should be [t, num_classes], where t is the number of tokens in the document. - """ import json diff --git a/src/hipool/models.py b/src/hipool/models.py index 619e240..78ccb3f 100644 --- a/src/hipool/models.py +++ b/src/hipool/models.py @@ -3,7 +3,6 @@ b: batch_size k: longest sequence length (in number of chunks) c: chunk length (in tokens) - """ import networkx as nx