Skip to content

Commit

Permalink
add __all__ for each file
Browse files Browse the repository at this point in the history
  • Loading branch information
geetu040 committed Dec 21, 2024
1 parent d9d3a49 commit 8f4c61f
Show file tree
Hide file tree
Showing 4 changed files with 8 additions and 0 deletions.
2 changes: 2 additions & 0 deletions src/transformers/models/depth_pro/configuration_depth_pro.py
Original file line number Diff line number Diff line change
Expand Up @@ -168,3 +168,5 @@ def __init__(
self.scaled_images_ratios = scaled_images_ratios
self.scaled_images_overlap_ratios = scaled_images_overlap_ratios
self.scaled_images_feature_dims = scaled_images_feature_dims

__all__ = ["DepthProConfig"]
Original file line number Diff line number Diff line change
Expand Up @@ -407,3 +407,5 @@ def post_process_depth_estimation(
outputs["predicted_depth"].append(predicted_depth)

return outputs

__all__ = ["DepthProImageProcessor"]
Original file line number Diff line number Diff line change
Expand Up @@ -380,3 +380,5 @@ def post_process_depth_estimation(
outputs["predicted_depth"].append(predicted_depth)

return outputs

__all__ = ["DepthProImageProcessorFast"]
2 changes: 2 additions & 0 deletions src/transformers/models/depth_pro/modeling_depth_pro.py
Original file line number Diff line number Diff line change
Expand Up @@ -1674,3 +1674,5 @@ def forward(
hidden_states=depth_pro_outputs.hidden_states,
attentions=depth_pro_outputs.attentions,
)

__all__ = ["DepthProPreTrainedModel", "DepthProModel", "DepthProForDepthEstimation"]

0 comments on commit 8f4c61f

Please sign in to comment.