Removed obsolete test

mkranzlein · Dec 7, 2023 · af0e892 · af0e892
1 parent 28405aa
commit af0e892
Showing 1 changed file with 2 additions and 8 deletions.
diff --git a/tests/curiam_reader_test.py b/tests/curiam_reader_test.py
@@ -1,5 +1,5 @@
 import pytest
-import torch
+
 from transformers import BertTokenizer
 
 from hipool import curiam_reader
@@ -8,7 +8,7 @@
 @pytest.fixture
 def curiam_sample():
     bert_tokenizer = BertTokenizer.from_pretrained('bert-base-uncased', do_lower_case=True)
-    dataset = curiam_reader.CuriamDataset(
+    dataset = curiam_reader.DocDataset(
         json_file_path="fixtures/data/curiam_sample.json",
         tokenizer=bert_tokenizer,
         chunk_len=50,
@@ -24,9 +24,3 @@ def test_read_json_len(curiam_sample):
 def test_read_json_tokens(curiam_sample):
     assert curiam_sample.documents[0][0] == "Justice"
     assert curiam_sample.documents[0][1] == "GORSUCH"
-
-
-def test_getitem_labels(curiam_sample):
-    first_chunk_labels = curiam_sample[0]["targets"][0]
-    first_token_labels = first_chunk_labels[0]
-    assert torch.equal(first_token_labels, torch.zeros((9)))