RoyAalekh commited on
Commit
5e91380
·
1 Parent(s): e206120

fix: Convert date to datetime for Override timestamp parameter

Browse files
Files changed (1) hide show
  1. 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
  )