add edt policy

ruiheng123 · Jul 1, 2024 · 485264b · 485264b
1 parent 1acb51d
commit 485264b
Show file tree

Hide file tree

Showing 6 changed files with 680 additions and 145 deletions.
diff --git a/ding/entry/tests/test_serial_entry.py b/ding/entry/tests/test_serial_entry.py
@@ -651,7 +651,9 @@ def test_discrete_dt():
         from ding.data import create_dataset
         from ding.config import compile_config
         from ding.model import DecisionTransformer
+        from ding.model.template.elastic_decision_transformer import ElasticDecisionTransformer
         from ding.policy import DTPolicy
+        from ding.policy.edt import EDTPolicy
         from ding.framework.middleware import interaction_evaluator, trainer, CkptSaver, \
             OfflineMemoryDataFetcher, offline_logger, termination_checker
         ding_init(config[0])