Merge branch 'main' of https://github.com/sisl/InteractionImitation into main

tests/nets/test_deepsets.py

@@ -41,7 +41,7 @@ def test_phi():
     y = phi(torch.rand(input_dim))
     y = phi(torch.rand(7,7,7,input_dim))
 
-    assert len(phi.parameters() > 0)
+    assert len(list(phi.parameters())) > 0
 
 def test_deepsets():
     m = ds.DeepSetsModule.from_config(ds_config)

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