Skip to content

Commit

Permalink
[CLEAN] ruff format
Browse files Browse the repository at this point in the history
  • Loading branch information
markkua committed May 17, 2024
1 parent 5126211 commit 9af23c9
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 4 deletions.
2 changes: 1 addition & 1 deletion infer.py
Original file line number Diff line number Diff line change
Expand Up @@ -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(
Expand Down
4 changes: 1 addition & 3 deletions marigold/marigold_pipeline.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)

Expand Down

0 comments on commit 9af23c9

Please sign in to comment.