Spaces:
Running
Running
Update landmarkdiff/safety.py to v0.3.2
Browse files- 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.
|
| 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,
|