Coverage for C: \ Users \ peaco \ OneDrive \ Documents \ GitHub \ mt_metadata \ mt_metadata \ processing \ aurora \ estimator.py: 100%

1# ===================================================== 

2# Imports 

3# ===================================================== 

4from typing import Annotated 

5 

6from pydantic import Field 

7 

8from mt_metadata.base import MetadataBase 

9from mt_metadata.common.enumerations import StrEnumerationBase 

10 

11 

12# ===================================================== 

13class EngineEnum(StrEnumerationBase): 

14 RME_RR = "RME_RR" 

15 RME = "RME" 

16 other = "other" 

17 

18 

19class Estimator(MetadataBase): 

20 engine: Annotated[ 

21 EngineEnum, 

22 Field( 

23 default=EngineEnum.RME_RR, 

24 description="The transfer function estimator engine", 

25 alias=None, 

26 json_schema_extra={ 

27 "units": None, 

28 "required": True, 

29 "examples": ["RME_RR"], 

30 }, 

31 ), 

32 ] 

33 

34 estimate_per_channel: Annotated[ 

35 bool, 

36 Field( 

37 default=True, 

38 description="Estimate per channel", 

39 alias=None, 

40 json_schema_extra={ 

41 "units": None, 

42 "required": True, 

43 "examples": ["True"], 

44 }, 

45 ), 

46 ]