[Test] adding mdcpdp to pytest and running configs

ai4co · Apr 5, 2024 · 9d0d743 · 9d0d743
1 parent 6fa342d
commit 9d0d743
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/configs/env/mdcpdp.yaml b/configs/env/mdcpdp.yaml
@@ -0,0 +1,14 @@
+_target_: rl4co.envs.MDCPDPEnv
+name: mdcpdp
+
+num_loc: 20
+num_depot: 4
+min_loc: 0
+max_loc: 1
+min_capacity: 1
+max_capacity: 2
+depot_mode: multiple
+
+data_dir: ${paths.root_dir}/data/mdcpdp
+val_file: mdcpdp${env.num_loc}_val_seed4321.npz
+test_file: mdcpdp${env.num_loc}_test_seed1234.npz
diff --git a/tests/test_envs.py b/tests/test_envs.py
@@ -20,6 +20,7 @@
     SPCTSPEnv,
     SVRPEnv,
     TSPEnv,
+    MDCPDPEnv,
 )
 from rl4co.models.nn.utils import random_policy, rollout
 
@@ -42,6 +43,7 @@
         PDPEnv,
         MTSPEnv,
         ATSPEnv,
+        MDCPDPEnv,
     ],
 )
 def test_routing(env_cls, batch_size=2, size=20):