[#2] Some simple cleanup

idiomify/builders.py

@@ -83,5 +83,3 @@ class TargetsBuilder(TensorBuilder):
         ], return_tensors="pt", add_special_tokens=False, padding=True, truncation=True)
         tgts = encodings['input_ids']
         return tgts  # (N, L)
-
-

idiomify/datamodules.py

@@ -1,7 +1,7 @@
-import pandas as pd
 import torch
-from typing import Tuple, Optional
+import pandas as pd
 from torch.utils.data import Dataset, DataLoader
+from typing import Tuple, Optional
 from pytorch_lightning import LightningDataModule
 from wandb.sdk.wandb_run import Run
 from idiomify.fetchers import fetch_literal2idiomatic

idiomify/pipeline.py

@@ -1,9 +1,5 @@
-
-# for inference
 from typing import List
-
 from transformers import BartTokenizer
-
 from builders import SourcesBuilder
 from models import Idiomifier