Introduce more DL/TW sampling methods

leonlan · Sep 19, 2023 · 036efd5 · 036efd5
1 parent 688fda8
commit 036efd5
Showing 1 changed file with 4 additions and 7 deletions.
diff --git a/sampling/__init__.py b/sampling/__init__.py
@@ -5,11 +5,8 @@
 from .SamplingMethod import SamplingMethod
 
 SAMPLING_METHODS: dict[str, SamplingMethod] = {
-    "DL2": partial(custom_time_windows, tw_type="deadlines", tw_width=2),
-    "DL4": partial(custom_time_windows, tw_type="deadlines", tw_width=4),
-    "DL8": partial(custom_time_windows, tw_type="deadlines", tw_width=8),
-    "TW2": partial(custom_time_windows, tw_type="time_windows", tw_width=2),
-    "TW4": partial(custom_time_windows, tw_type="time_windows", tw_width=4),
-    "TW8": partial(custom_time_windows, tw_type="time_windows", tw_width=8),
-    "euro_neurips": euro_neurips,
+    f"{name}{tw_width}": partial(custom_time_windows, tw_type, tw_width)
+    for name, tw_type in [("DL", "deadlines"), ("TW", "time_windows")]
+    for tw_width in range(1, 9)
 }
+SAMPLING_METHODS["euro_neurips"] = euro_neurips