diff --git a/scripts/utils.py b/scripts/utils.py index 15edffc..f0d7f53 100644 --- a/scripts/utils.py +++ b/scripts/utils.py @@ -377,12 +377,14 @@ def sort_instances(df, sort_by, preferred=None): """ try: - df = df.sort_values(by=sort_by) if preferred: df['priority'] = (df[sort_by] == preferred).astype(int) df = df.sort_values(by=['priority', sort_by], ascending=[False, True]).drop(columns='priority') + else: + df = df.sort_values(by=sort_by) + except Exception as err: print_colors('[-] Sorting failed',is_error=True)