Save centroids and embeddings in a dictionary with their "feature_name" as the key. This cleans up some code in _get_cost_matrix as well.
Refactor _check_feature to _validate_feature_name for clarity.
_calc_embeddings is passed the input data instead of relying on class attributes. This function could be moved out of the class itself.
Various whitespace changes for better readability.
Update _get_parent to use track_ids NOT indices of the predictions. Tests are failing rarely because it is possible that a track is not included in the prediction output, so the indices and track_ids would be off.
centroidsandembeddingsin a dictionary with their "feature_name" as the key. This cleans up some code in_get_cost_matrixas well._check_featureto_validate_feature_namefor clarity._calc_embeddingsis passed the input data instead of relying on class attributes. This function could be moved out of the class itself._get_parentto use track_ids NOT indices of the predictions. Tests are failing rarely because it is possible that a track is not included in the prediction output, so the indices and track_ids would be off.