Coverage for docs_src/parameter_types/enum/tutorial002_an.py: 100%

1from enum import Enum 9 ctx1iabcdefgh

2 

3import typer 9 ctx1iabcdefgh

4from typing_extensions import Annotated 9 ctx1iabcdefgh

5 

6 

7class NeuralNetwork(str, Enum): 9 ctx1iabcdefgh

8 simple = "simple" 9 ctx1iabcdefgh

9 conv = "conv" 9 ctx1iabcdefgh

10 lstm = "lstm" 9 ctx1iabcdefgh

11 

12 

13def main( 8 ctx1abcdefgh

14 network: Annotated[ 

15 NeuralNetwork, typer.Option(case_sensitive=False) 

16 ] = NeuralNetwork.simple, 

17): 

18 print(f"Training neural network of type: {network.value}") 9 ctx1iabcdefgh

19 

20 

21if __name__ == "__main__": 9 ctx1iabcdefgh

22 typer.run(main) 9 ctx1iabcdefgh