From 8d702ceccdb18047efdf86a1ec791862f6a23aa3 Mon Sep 17 00:00:00 2001 From: Adonai Vera <45982251+AdonaiVera@users.noreply.github.com> Date: Wed, 31 Jan 2024 02:21:46 -0500 Subject: [PATCH] Remove csv file --- supervision/utils/file.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/supervision/utils/file.py b/supervision/utils/file.py index a827ad18..46299101 100644 --- a/supervision/utils/file.py +++ b/supervision/utils/file.py @@ -1,5 +1,4 @@ import json -import csv from pathlib import Path from typing import List, Optional, Union, Dict, Any from supervision.detection.core import Detections @@ -70,7 +69,7 @@ class JSONSink: detection_data[key] = value[i] if hasattr(value, '__getitem__') else value if custom_data: detection_data.update(custom_data) - + self.data.append(detection_data) class NumpyJsonEncoder(json.JSONEncoder):