[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
choderalab · Nov 12, 2024 · a0d0eb4 · a0d0eb4
1 parent cb3dac7
commit a0d0eb4
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/missense_kinase_toolkit/ml/src/esm2/inference.py b/missense_kinase_toolkit/ml/src/esm2/inference.py
@@ -1,6 +1,6 @@
+import matplotlib.pyplot as plt
 import numpy as np
 import pandas as pd
-import matplotlib.pyplot as plt
 import seaborn as sns
 from transformers import AutoTokenizer, EsmForSequenceClassification
 

diff --git a/missense_kinase_toolkit/ml/src/esm2/utils.py b/missense_kinase_toolkit/ml/src/esm2/utils.py
@@ -1,11 +1,12 @@
 import argparse
 import datetime
 import os
+
+import matplotlib.pyplot as plt
 import numpy as np
 import pandas as pd
 from datasets import Dataset, load_dataset
 from sklearn.metrics import mean_squared_error
-import matplotlib.pyplot as plt
 
 
 def parsearg_utils():
@@ -234,7 +235,7 @@ def save_csv2csv(
     -------
     None
     """
-    df = df.loc[df["Mutant"].apply(lambda x: x is False), ].reset_index(drop=True)
+    df = df.loc[df["Mutant"].apply(lambda x: x is False),].reset_index(drop=True)
     df_shuffle = df.copy().sample(frac=1, random_state=seed).reset_index(drop=True)
     df_out = df_shuffle[[col_seq, col_lab]]
     df_out.columns = ["seq", "label"]