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

1from enum import Enum 7 ctx1abcdefg

2 

3import typer 7 ctx1abcdefg

4 

5 

6class NeuralNetwork(str, Enum): 7 ctx1abcdefg

7 simple = "simple" 7 ctx1abcdefg

8 conv = "conv" 7 ctx1abcdefg

9 lstm = "lstm" 7 ctx1abcdefg

10 

11 

12app = typer.Typer() 7 ctx1abcdefg

13 

14 

15@app.command() 7 ctx1abcdefg

16def main( 7 ctx1abcdefg

17 network: NeuralNetwork = typer.Option(NeuralNetwork.simple, case_sensitive=False), 

18): 

19 print(f"Training neural network of type: {network.value}") 7 ctx1abcdefg

20 

21 

22if __name__ == "__main__": 7 ctx1abcdefg

23 app() 7 ctx1abcdefg