Spaces:
Running
Running
fix: Convert date to datetime for Override timestamp parameter
Browse files- rl/training.py +2 -2
rl/training.py
CHANGED
|
@@ -7,7 +7,7 @@ case prioritization policies through simulation-based rewards.
|
|
| 7 |
import numpy as np
|
| 8 |
from pathlib import Path
|
| 9 |
from typing import List, Tuple, Dict, Optional
|
| 10 |
-
from datetime import date, timedelta
|
| 11 |
import random
|
| 12 |
|
| 13 |
from scheduler.data.case_generator import CaseGenerator
|
|
@@ -135,7 +135,7 @@ class RLTrainingEnvironment:
|
|
| 135 |
override_type=OverrideType.PRIORITY,
|
| 136 |
case_id=case.case_id,
|
| 137 |
judge_id="RL-JUDGE",
|
| 138 |
-
timestamp=self.current_date,
|
| 139 |
new_priority=case.get_priority_score() + priority_boost,
|
| 140 |
)
|
| 141 |
)
|
|
|
|
| 7 |
import numpy as np
|
| 8 |
from pathlib import Path
|
| 9 |
from typing import List, Tuple, Dict, Optional
|
| 10 |
+
from datetime import date, datetime, timedelta
|
| 11 |
import random
|
| 12 |
|
| 13 |
from scheduler.data.case_generator import CaseGenerator
|
|
|
|
| 135 |
override_type=OverrideType.PRIORITY,
|
| 136 |
case_id=case.case_id,
|
| 137 |
judge_id="RL-JUDGE",
|
| 138 |
+
timestamp=datetime.combine(self.current_date, datetime.min.time()),
|
| 139 |
new_priority=case.get_priority_score() + priority_boost,
|
| 140 |
)
|
| 141 |
)
|