bugfix in deepsets

2021-07-22 18:44:12 +02:00
parent 91d052445e
commit 1c22bd6111
4 changed files with 8 additions and 3 deletions
--- a/tests/policies/test_policies.py
+++ b/tests/policies/test_policies.py
@@ -25,4 +25,6 @@ def test_deepsets_policy():
        "path_y": path_y,
    }
    
-    module(sample)
+    module(sample)
+
+    # TODO: Refine test to also test for batch data