Skip to content

Commit

Permalink
Format code with black and isort (#47)
Browse files Browse the repository at this point in the history
This commit fixes the style issues introduced in fe62d8b according to the output
from black and isort.

Details: https://deepsource.io/gh/AtomScott/SoccerTrack/transform/55d99477-e19f-4c8c-9022-a5938920150b/

Co-authored-by: deepsource-autofix[bot] <62050782+deepsource-autofix[bot]@users.noreply.github.com>
  • Loading branch information
deepsource-autofix[bot] authored Jan 14, 2023
1 parent fe62d8b commit cfe04be
Show file tree
Hide file tree
Showing 7 changed files with 18 additions and 20 deletions.
10 changes: 5 additions & 5 deletions soccertrack/metrics/__init__.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,8 @@
from soccertrack.metrics.cost_matrix_metrics import (
BaseCostMatrixMetric,
CosineCMM,
IoUCMM,
)
from soccertrack.metrics.hota import hota_score
from soccertrack.metrics.identity import identity_score
from soccertrack.metrics.mota import mota_score
Expand All @@ -8,11 +13,6 @@
map_score,
map_score_range,
)
from soccertrack.metrics.cost_matrix_metrics import (
BaseCostMatrixMetric,
IoUCMM,
CosineCMM,
)

__all__ = [
"hota_score",
Expand Down
6 changes: 1 addition & 5 deletions soccertrack/metrics/cost_matrix_metrics.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,13 +6,9 @@
import numpy as np
from scipy.spatial.distance import cdist

from soccertrack.checks import _check_cost_matrix, _check_detections, _check_trackers
from soccertrack.metrics import iou_score
from soccertrack.types import Detection, Tracker
from soccertrack.checks import (
_check_trackers,
_check_detections,
_check_cost_matrix,
)


class BaseCostMatrixMetric(ABC):
Expand Down
1 change: 1 addition & 0 deletions soccertrack/metrics/tracking_preprocess.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,7 @@

import numpy as np
from scipy.spatial.distance import cdist

from soccertrack.metrics.object_detection import convert_to_x1y1x2y2, iou_score


Expand Down
11 changes: 7 additions & 4 deletions soccertrack/tracking_model/matching.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,10 +10,13 @@
from scipy.optimize import linear_sum_assignment
from scipy.spatial.distance import cdist

from soccertrack.checks import (_check_cost_matrix, _check_detections,
_check_matches, _check_trackers)
from soccertrack.metrics import (BaseCostMatrixMetric, CosineCMM, IoUCMM,
iou_score)
from soccertrack.checks import (
_check_cost_matrix,
_check_detections,
_check_matches,
_check_trackers,
)
from soccertrack.metrics import BaseCostMatrixMetric, CosineCMM, IoUCMM, iou_score
from soccertrack.tracking_model import SingleObjectTracker
from soccertrack.types import Detection

Expand Down
3 changes: 1 addition & 2 deletions soccertrack/tracking_model/multi_object_tracker.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,8 +12,7 @@

from soccertrack import BBoxDataFrame, Camera
from soccertrack.logger import logger, tqdm
from soccertrack.tracking_model.tracker import (KalmanTracker,
SingleObjectTracker)
from soccertrack.tracking_model.tracker import KalmanTracker, SingleObjectTracker
from soccertrack.types import Box, Detection, Track, Vector, _pathlike

DEFAULT_MODEL_SPEC = ModelPreset.constant_velocity_and_static_box_size_2d.value
Expand Down
4 changes: 2 additions & 2 deletions soccertrack/tracking_model/single_object_tracker.py
Original file line number Diff line number Diff line change
@@ -1,16 +1,16 @@
from __future__ import annotations

import uuid
from abc import ABC, abstractmethod
from collections.abc import Iterable
from typing import Any, Callable, Dict, List, Optional, Sequence, Tuple, Type, Union
from abc import ABC, abstractmethod

import numpy as np
import pandas as pd

from soccertrack import BBoxDataFrame
from soccertrack.logger import logger
from soccertrack.types import Detection, Box, Vector, Tracker
from soccertrack.types import Box, Detection, Tracker, Vector


class SingleObjectTracker(Tracker):
Expand Down
3 changes: 1 addition & 2 deletions tests/test_tracking_model/test_matching.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,8 +16,7 @@
import numpy as np

from soccertrack.logger import *
from soccertrack.tracking_model import (MotionVisualMatchingFunction,
SingleObjectTracker)
from soccertrack.tracking_model import MotionVisualMatchingFunction, SingleObjectTracker
from soccertrack.types import Detection

dets = [
Expand Down

0 comments on commit cfe04be

Please sign in to comment.