Merge pull request #30 from calico/fix-black-format

fix black format
calico · May 8, 2024 · bfcee4d · bfcee4d
2 parents 8ab0462 + 47a94d1
commit bfcee4d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/baskerville/dataset.py b/src/baskerville/dataset.py
@@ -377,7 +377,7 @@ def untransform_preds(preds, targets_df, unscale=False, unclip=True):
       preds (np.array): Untransformed predictions LxT.
     """
     # clip soft
-    if unclip :
+    if unclip:
         cs = np.expand_dims(np.array(targets_df.clip_soft), axis=0)
         preds_unclip = cs - 1 + (preds - cs + 1) ** 2
         preds = np.where(preds > cs, preds_unclip, preds)
@@ -409,7 +409,7 @@ def untransform_preds1(preds, targets_df, unscale=False, unclip=True):
     preds = preds / scale
 
     # clip soft
-    if unclip :
+    if unclip:
         cs = np.expand_dims(np.array(targets_df.clip_soft), axis=0)
         preds_unclip = cs + (preds - cs) ** 2
         preds = np.where(preds > cs, preds_unclip, preds)