Skip to content

Commit 9b7385c

Browse files
committed
Add doctest to main() function per reviewer request
1 parent 44c8f96 commit 9b7385c

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

computer_vision/vision_transformer_demo.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -176,6 +176,14 @@ def main() -> None:
176176
Main function demonstrating Vision Transformer usage.
177177
178178
Downloads a sample image and performs classification.
179+
180+
Examples:
181+
>>> # Verify main is callable
182+
>>> callable(main)
183+
True
184+
>>> # Verify main returns None
185+
>>> main() is None # doctest: +SKIP
186+
True
179187
"""
180188
print("Vision Transformer (ViT) Image Classification Demo")
181189
print("=" * 60)

0 commit comments

Comments
 (0)