Fix dataclass import.

poldeepner2/model/hf_for_token_calssification.py

@@ -5,8 +5,7 @@ import yaml
 from typing import List
 
 import torch
-from attr import dataclass
-from dataclasses import field
+from dataclasses import field, dataclass
 from transformers import AutoModel, AutoTokenizer, AutoConfig
 import torch.nn as nn
 import torch.nn.functional as F

poleval_ner_test_v2.py

@@ -4,7 +4,7 @@ import getopt
 import json
 import sys
 
-from attr import dataclass
+from dataclasses import dataclass
 from dateutil import parser
 from tqdm import tqdm