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)