From 9af23c9f3b941c4457e67a36960aa485d60ecafa Mon Sep 17 00:00:00 2001 From: Bingxin Date: Fri, 17 May 2024 22:47:17 +0200 Subject: [PATCH] [CLEAN] ruff format --- infer.py | 2 +- marigold/marigold_pipeline.py | 4 +--- 2 files changed, 2 insertions(+), 4 deletions(-) diff --git a/infer.py b/infer.py index 6f95209..51e8ddf 100644 --- a/infer.py +++ b/infer.py @@ -216,7 +216,7 @@ def check_directory(directory): logging.info( f"{pipe.prediction_type = }, {pipe.scale_invariant = }, {pipe.shift_invariant = }" ) - + # -------------------- Inference and saving -------------------- with torch.no_grad(): for batch in tqdm( diff --git a/marigold/marigold_pipeline.py b/marigold/marigold_pipeline.py index 970f126..522ae77 100644 --- a/marigold/marigold_pipeline.py +++ b/marigold/marigold_pipeline.py @@ -292,9 +292,7 @@ def __call__( if self.scale_invariant: max_d = torch.max(depth_pred) else: - raise NotImplementedError( - "Metric depth is not supported." - ) + raise NotImplementedError("Metric depth is not supported.") depth_pred = (depth_pred - min_d) / (max_d - min_d)