Spaces:
Runtime error
Runtime error
Update tests/test_inference.py
Browse files- tests/test_inference.py +1 -1
tests/test_inference.py
CHANGED
|
@@ -8,4 +8,4 @@ def test_inference():
|
|
| 8 |
model = YOLOv8Model()
|
| 9 |
result = model.predict(frame)
|
| 10 |
assert isinstance(result, pd.DataFrame), "Expected results in DataFrame format"
|
| 11 |
-
assert "name" in result.columns, "Violation class (name) missing in results"
|
|
|
|
| 8 |
model = YOLOv8Model()
|
| 9 |
result = model.predict(frame)
|
| 10 |
assert isinstance(result, pd.DataFrame), "Expected results in DataFrame format"
|
| 11 |
+
assert "name" in result.columns, "Violation class (name) missing in results"
|