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

1from enum import Enum 9 ctx1iabcdefgh

2 

3import typer 9 ctx1iabcdefgh

4 

5 

6class NeuralNetwork(str, Enum): 9 ctx1iabcdefgh

7 simple = "simple" 9 ctx1iabcdefgh

8 conv = "conv" 9 ctx1iabcdefgh

9 lstm = "lstm" 9 ctx1iabcdefgh

10 

11 

12def main( 8 ctx1abcdefgh

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

14): 

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

16 

17 

18if __name__ == "__main__": 9 ctx1iabcdefgh

19 typer.run(main) 9 ctx1iabcdefgh