Skip to content

Commit

Permalink
cleanup
Browse files Browse the repository at this point in the history
  • Loading branch information
ArneBinder committed Sep 16, 2024
1 parent 904fac9 commit c76caca
Showing 1 changed file with 1 addition and 5 deletions.
6 changes: 1 addition & 5 deletions src/pie_modules/taskmodules/cross_text_binary_coref.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,5 @@
import copy
import logging
from functools import partial
from typing import (
Any,
Dict,
Expand All @@ -20,14 +19,11 @@
from pytorch_ie.annotations import Span
from pytorch_ie.core import TaskEncoding, TaskModule
from pytorch_ie.utils.window import get_window_around_slice
from torch.nn.functional import threshold
from torchmetrics import ClasswiseWrapper, F1Score, Metric, MetricCollection
from torchmetrics import MetricCollection
from torchmetrics.classification import (
BinaryAUROC,
BinaryAveragePrecision,
BinaryF1Score,
BinaryPrecisionRecallCurve,
BinaryROC,
)
from transformers import AutoTokenizer, BatchEncoding
from typing_extensions import TypeAlias
Expand Down

0 comments on commit c76caca

Please sign in to comment.