o wiD @s`dZddlmZmZmZmZmZddlmZddl m Z GdddeZ eGdddeZ d S) z/Telemetry configuration protocol for OneLogger.)DictOptionalProtocolUnionruntime_checkable)StrEnum)AttributeValuec@s$eZdZdZdZdZdZdZdZdS)ApplicationTypez"Enum for common application types.training validationbatch_inferenceonline_inferencedata_processingN) __name__ __module__ __qualname____doc__TRAINING VALIDATIONBATCH_INFERENCEONLINE_INFERENCEDATA_PROCESSINGrr_/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/nv_one_logger/api/telemetry_config.pyr sr c@seZdZdZedefddZedefddZedefddZedefd d Z ede e e ffd d Z edeee effd dZdS)TelemetryConfigaProtocol for telemetry configuration. This protocol defines the interface that any telemetry configuration must implement. It allows the core module to reference telemetry configs without creating circular dependencies and provides type safety while allowing different implementations for different application types. The @runtime_checkable decorator enables runtime type checking with isinstance() and issubclass(). Without this decorator, isinstance(obj, TelemetryConfig) would always return False, even for objects that structurally implement all the required methods and properties. Example usage: # This works because of @runtime_checkable if isinstance(some_config, TelemetryConfig): print("Valid telemetry config") Note: @runtime_checkable has some limitations - it only checks for the presence of methods and properties, not their signatures or return types. It's a structural check, not a behavioral one. returncCdS)zXWhether the application has training iterations that OneLogger should track metrics for.Nrselfrrris_train_iterations_enabled5z+TelemetryConfig.is_train_iterations_enabledcCr)zeWhether the application has validation/evaluation iterations that OneLogger should track metrics for.Nrrrrr is_validation_iterations_enabled;r z0TelemetryConfig.is_validation_iterations_enabledcCr)zTWhether the application has test iterations that OneLogger should track metrics for.Nrrrrris_test_iterations_enabledAr z*TelemetryConfig.is_test_iterations_enabledcCr)zRWhether the application saves checkpoints that OneLogger should track metrics for.Nrrrrris_save_checkpoint_enabledGr z*TelemetryConfig.is_save_checkpoint_enabledcCr)z:Get the application type for this telemetry configuration.Nrrrrrapp_typeMr zTelemetryConfig.app_typecCr)z4Custom metadata specific to telemetry configuration.Nrrrrrcustom_metadataTr zTelemetryConfig.custom_metadataN)rrrrpropertyboolrr!r"r#rr strr$rrrr%rrrrrs rN) rtypingrrrrrstrenumrnv_one_logger.core.attributesrr rrrrrs