Fix tests

2021-07-26 16:35:45 +02:00
parent 69359b5af3
commit 5e89a83c88
4 changed files with 10 additions and 10 deletions
--- a/tests/policies/test_policies.py
+++ b/tests/policies/test_policies.py
@@ -7,7 +7,7 @@ with open(config_path, 'r') as cfg:
    config = json5.load(cfg)

 def test_deepsets_policy():
-    module = DeepSetsPolicy(config["ego_state"], config["deepsets"], config["path_encoder"], config["head"])
+    module = DeepSetsPolicy(config)
    
    ns = 5 
    nv = 7