Add test and example config for DeepSetsPolicy

tests/policies/test_policies.py

@@ -0,0 +1,20 @@
+import torch
+from src.policies.policy import DeepSetsPolicy
+import json5
+
+config_path = "config/networks.json5"
+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"])
+    
+    ns = 5 
+    nv = 7
+    npath = 20
+
+    ego_state = torch.rand(ns)
+    relative_state = torch.rand(nv, ns)
+    path = torch.rand(npath, 2)
+    
+    module(ego_state, relative_state, path)