subhdotsol commited on
Commit
40e374a
·
1 Parent(s): f15649d

refactor: replace datetime.utcnow() with timezone-aware datetime.now(UTC)

Browse files
Files changed (1) hide show
  1. server/environment.py +2 -2
server/environment.py CHANGED
@@ -1,7 +1,7 @@
1
  import uuid
2
  import logging
3
  from typing import Optional, Callable
4
- from datetime import datetime
5
 
6
  from models import (
7
  AttackAction,
@@ -37,7 +37,7 @@ class RedTeamEnvironment:
37
  self.attack_history = []
38
  self.episode_id = f"ep_{uuid.uuid4().hex[:12]}"
39
  self.is_active = True
40
- self.created_at = datetime.utcnow()
41
 
42
  if self.reward_computer and hasattr(self.reward_computer, "reset"):
43
  self.reward_computer.reset()
 
1
  import uuid
2
  import logging
3
  from typing import Optional, Callable
4
+ from datetime import datetime, timezone
5
 
6
  from models import (
7
  AttackAction,
 
37
  self.attack_history = []
38
  self.episode_id = f"ep_{uuid.uuid4().hex[:12]}"
39
  self.is_active = True
40
+ self.created_at = datetime.now(timezone.utc)
41
 
42
  if self.reward_computer and hasattr(self.reward_computer, "reset"):
43
  self.reward_computer.reset()