PrashanthB461 commited on
Commit
b19faa8
·
verified ·
1 Parent(s): 1c7627f

Update tests/test_inference.py

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