dreamlessx commited on
Commit
ded6c17
·
verified ·
1 Parent(s): bb003e6

Update landmarkdiff/safety.py to v0.3.2

Browse files
Files changed (1) hide show
  1. landmarkdiff/safety.py +21 -1
landmarkdiff/safety.py CHANGED
@@ -41,6 +41,26 @@ class SafetyResult:
41
  checks: dict[str, bool] = field(default_factory=dict)
42
  details: dict[str, object] = field(default_factory=dict)
43
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
44
  def add_failure(self, name: str, message: str) -> None:
45
  self.passed = False
46
  self.failures.append(message)
@@ -66,7 +86,7 @@ class SafetyValidator:
66
 
67
  def __init__(
68
  self,
69
- identity_threshold: float = 0.6,
70
  max_displacement_fraction: float = 0.05,
71
  min_face_confidence: float = 0.5,
72
  max_yaw_degrees: float = 45.0,
 
41
  checks: dict[str, bool] = field(default_factory=dict)
42
  details: dict[str, object] = field(default_factory=dict)
43
 
44
+ def __repr__(self) -> str:
45
+ return (
46
+ f"SafetyResult(passed={self.passed}, "
47
+ f"failures={self.failures}, "
48
+ f"warnings={self.warnings}, "
49
+ f"checks={self.checks}, "
50
+ f"details={self.details})"
51
+ )
52
+
53
+ def __eq__(self, other: object) -> bool:
54
+ if not isinstance(other, SafetyResult):
55
+ return NotImplemented
56
+ return (
57
+ self.passed == other.passed
58
+ and self.failures == other.failures
59
+ and self.warnings == other.warnings
60
+ and self.checks == other.checks
61
+ and self.details == other.details
62
+ )
63
+
64
  def add_failure(self, name: str, message: str) -> None:
65
  self.passed = False
66
  self.failures.append(message)
 
86
 
87
  def __init__(
88
  self,
89
+ identity_threshold: float = 0.5,
90
  max_displacement_fraction: float = 0.05,
91
  min_face_confidence: float = 0.5,
92
  max_yaw_degrees: float = 45.0,