Adding features to label tensor class (#29)

* adding label.setter for runtime check on labels
This commit is contained in:
Dario Coscia
2022-11-04 17:26:05 +01:00
committed by GitHub
parent d06f28de7b
commit a92a764844
2 changed files with 21 additions and 2 deletions

View File

@@ -20,6 +20,8 @@ def test_labels():
tensor = LabelTensor(data, labels)
assert isinstance(tensor, torch.Tensor)
assert tensor.labels == labels
with pytest.raises(ValueError):
tensor.labels = labels[:-1]
def test_extract():