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_forward_pass.py
+++ b/tests/policies/test_forward_pass.py
@@ -9,6 +9,9 @@ with open(config_path, 'r') as cfg:

 filestr = 'tests/policies/base_'

+
 batch = pickle.load(open(filestr+'_test_batch.pkl', 'rb'))
+batch['relative_state'] = batch['relative_state'].float()
 policy = BehaviorCloningPolicy.load_model(config, filestr)
+policy.policy = policy.policy.type(batch['state'].dtype)
 action = policy(batch)