From 47a94d159d0576a402741b84fcd1edf8e7820b41 Mon Sep 17 00:00:00 2001 From: lruizcalico Date: Wed, 8 May 2024 16:04:46 -0700 Subject: [PATCH] fix black format --- src/baskerville/dataset.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/baskerville/dataset.py b/src/baskerville/dataset.py index 785e8cc..c8360c3 100644 --- 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)