fix old codes

2022-07-11 10:58:15 +02:00
parent 088649e042
commit f526a26050
19 changed files with 385 additions and 457 deletions
--- a/tests/test_label_tensor.py
+++ b/tests/test_label_tensor.py
@@ -72,9 +72,8 @@ def test_merge():

 def test_merge():
    tensor = LabelTensor(data, labels)
-    tensor_a = tensor.extract('a')
    tensor_b = tensor.extract('b')
    tensor_c = tensor.extract('c')

-    tensor_bb = tensor_b.append(tensor_b)
-    assert torch.allclose(tensor_b, tensor.extract(['b', 'c']))
+    tensor_bc = tensor_b.append(tensor_c)
+    assert torch.allclose(tensor_bc, tensor.extract(['b', 'c']))