test

aphp · May 15, 2024 · 722e4a8 · 722e4a8
1 parent ff538d6
commit 722e4a8
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 11 deletions.
diff --git a/foldedtensor/__init__.py b/foldedtensor/__init__.py
@@ -103,6 +103,10 @@ def backward(ctx, grad_output):
         return grad_input, None
 
 
+def this_is_a_test():
+    print("ok")
+
+
 type_to_dtype_dict = {
     int: torch.tensor([0]).dtype,
     float: torch.tensor([0.0]).dtype,

diff --git a/tests/test_folded_tensor.py b/tests/test_folded_tensor.py
@@ -406,14 +406,14 @@ def test_imbalanced_sequence_2():
     assert "'int' object is not iterable" in str(e.value)
 
 
-def test_max():
-    ft = as_folded_tensor(
-        [
-            [0, 1, 2],
-            [3, 4],
-        ],
-        dtype=torch.float,
-    )
-    values, indices = ft.max(-1)
-    assert (values == torch.tensor([2, 4])).all()
-    assert (indices == torch.tensor([2, 1])).all()
+# def test_max():
+#     ft = as_folded_tensor(
+#         [
+#             [0, 1, 2],
+#             [3, 4],
+#         ],
+#         dtype=torch.float,
+#     )
+#     values, indices = ft.max(-1)
+#     assert (values == torch.tensor([2, 4])).all()
+#     assert (indices == torch.tensor([2, 1])).all()