Unexclude constants folder (#34013)

* unexclude constants folder

* fix mypy errors

Author: CoderKevinZhang

sdk/ml/azure-ai-ml/azure/ai/ml/constants/_component.py

@@ -2,6 +2,8 @@
 # Copyright (c) Microsoft Corporation. All rights reserved.
 # ---------------------------------------------------------
 
+from typing import Dict
+
 DO_WHILE_MAX_ITERATION = 1000
 
 
@@ -106,7 +108,7 @@ class IOConstants:
         ComponentParameterTypes.NUMBER: "Number",
         ComponentParameterTypes.BOOLEAN: "Boolean",
     }
-    PARAM_PARSERS = {
+    PARAM_PARSERS: Dict = {
         ComponentParameterTypes.INTEGER: lambda v: int(float(v)),  # parse case like 10.0 -> 10
         ComponentParameterTypes.BOOLEAN: lambda v: str(v).lower() == "true",
         ComponentParameterTypes.NUMBER: float,

sdk/ml/azure-ai-ml/azure/ai/ml/entities/_assets/_artifacts/model.py

@@ -198,9 +198,7 @@ def _update_path(self, asset_artifact: ArtifactStorageInfo) -> None:
                 asset_artifact.relative_path,
             )
 
-    def _to_arm_resource_param(
-        self, **kwargs: Any  # pylint: disable=unused-argument
-    ) -> ArmConstants.MODEL_VERSION_TYPE:
+    def _to_arm_resource_param(self, **kwargs: Any) -> Dict:  # pylint: disable=unused-argument
         properties = self._to_rest_object().properties
 
         return {

sdk/ml/azure-ai-ml/azure/ai/ml/entities/_credentials.py

@@ -5,7 +5,7 @@
 # pylint: disable=protected-access,redefined-builtin
 
 from abc import ABC
-from typing import Dict, List, Optional, Union
+from typing import Any, Dict, List, Optional, Union
 
 from azure.ai.ml._azure_environments import _get_active_directory_url_from_metadata
 from azure.ai.ml._restclient.v2022_01_01_preview.models import Identity as RestIdentityConfiguration
@@ -66,7 +66,7 @@
 
 class _BaseIdentityConfiguration(ABC, DictMixin, RestTranslatableMixin):
     def __init__(self) -> None:
-        self.type = None
+        self.type: Any = None
 
 
 class AccountKeyConfiguration(RestTranslatableMixin, DictMixin):

sdk/ml/azure-ai-ml/azure/ai/ml/entities/_job/distribution.py

@@ -33,7 +33,7 @@ class DistributionConfiguration(RestTranslatableMixin):
     """
 
     def __init__(self, **kwargs: Any) -> None:
-        self.type = None
+        self.type: Any = None
 
     @classmethod
     def _from_rest_object(

sdk/ml/azure-ai-ml/azure/ai/ml/entities/_job/job_limits.py

@@ -24,7 +24,7 @@ class JobLimits(RestTranslatableMixin, ABC):
     def __init__(
         self,
     ) -> None:
-        self.type = None
+        self.type: Any = None
 
     def __eq__(self, other: Any) -> bool:
         if not isinstance(other, JobLimits):

sdk/ml/azure-ai-ml/azure/ai/ml/entities/_job/job_service.py

@@ -53,7 +53,7 @@ def __init__(  # pylint: disable=unused-argument
         **kwargs: Dict,
     ) -> None:
         self.endpoint = endpoint
-        self.type = type
+        self.type: Any = type
         self.nodes = nodes
         self.status = status
         self.port = port
@@ -109,7 +109,9 @@ def _to_rest_job_services(
     def _from_rest_job_service_object(cls, obj: RestJobService) -> "JobServiceBase":
         return cls(
             endpoint=obj.endpoint,
-            type=JobServiceTypeNames.REST_TO_ENTITY.get(obj.job_service_type, None) if obj.job_service_type else None,
+            type=JobServiceTypeNames.REST_TO_ENTITY.get(obj.job_service_type, None)  # type: ignore[arg-type]
+            if obj.job_service_type
+            else None,
             nodes="all" if obj.nodes else None,
             status=obj.status,
             port=obj.port,

sdk/ml/azure-ai-ml/azure/ai/ml/entities/_job/pipeline/_pipeline_expression.py

@@ -60,7 +60,7 @@ def _enumerate_operation_combination() -> Dict[str, Union[str, Exception]]:
       * An Exception
     :rtype: Dict[str, Union[str, Exception]]
     """
-    res = {}
+    res: Dict = {}
     primitive_types_values = {
         NONE_PARAMETER_TYPE: repr(None),
         ComponentParameterTypes.BOOLEAN: repr(True),

sdk/ml/azure-ai-ml/azure/ai/ml/entities/_job/queue_settings.py

@@ -77,7 +77,7 @@ def _validate(self) -> None:
                     error_category=ErrorCategory.USER_ERROR,
                     error_type=ValidationErrorType.INVALID_VALUE,
                 )
-            valid_keys = list(enum_class.ENTITY_TO_REST.keys())
+            valid_keys = list(enum_class.ENTITY_TO_REST.keys())  # type: ignore[attr-defined]
             if value and value.lower() not in valid_keys:
                 msg = f"{key} should be one of {valid_keys}, but received '{value}'."
                 raise ValidationException(

sdk/ml/azure-ai-ml/azure/ai/ml/entities/_monitoring/definition.py

@@ -102,7 +102,7 @@ def _from_rest_object(
         obj: RestMonitorDefinition,
         **kwargs: Any,
     ) -> "MonitorDefinition":
-        from_rest_alert_notification = None
+        from_rest_alert_notification: Any = None
         if obj.alert_notification_setting:
             if isinstance(obj.alert_notification_setting, AzMonMonitoringAlertNotificationSettings):
                 from_rest_alert_notification = AZMONITORING

sdk/ml/azure-ai-ml/azure/ai/ml/entities/_monitoring/input_data.py

@@ -38,8 +38,8 @@ class MonitorInputData(RestTranslatableMixin):
     def __init__(
         self,
         *,
-        type: MonitorInputDataType = None,
-        data_context: MonitorDatasetContext = None,
+        type: Optional[MonitorInputDataType] = None,
+        data_context: Optional[MonitorDatasetContext] = None,
         target_columns: Optional[Dict] = None,
         job_type: Optional[str] = None,
         uri: Optional[str] = None,