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_forward_pass.py

@@ -11,11 +11,11 @@ filestr = 'tests/policies/base_'
 
 
 def test_forward_pass():
-batch = pickle.load(open(filestr + '_test_batch.pkl', 'rb'))
+    batch = pickle.load(open(filestr + '_test_batch.pkl', 'rb'))
-batch['relative_state'] = batch['relative_state'].float()
+    batch['relative_state'] = batch['relative_state'].float()
-policy = BehaviorCloningPolicy.load_model(config, filestr)
+    policy = BehaviorCloningPolicy.load_model(config, filestr)
-policy.policy = policy.policy.type(batch['state'].dtype)
+    policy.policy = policy.policy.type(batch['state'].dtype)
-action = policy(batch)
+    action = policy(batch)
-i = torch.where(action.isnan())[0]
+    i = torch.where(action.isnan())[0]
-for key in batch.keys():
+    for key in batch.keys():
         print(batch[key][i])

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