Unexclude sweep folder (#34016)

* unexclude sweep folder

* update

* update

Author: CoderKevinZhang

sdk/ml/azure-ai-ml/azure/ai/ml/automl/_automl_image.py

@@ -49,7 +49,7 @@ def _create_image_job(
     :return: An AutoML Image Job
     :rtype: TImageJob
     """
-    image_job = job_cls(primary_metric=primary_metric, **kwargs)
+    image_job = job_cls(primary_metric=primary_metric, **kwargs)  # type: ignore[operator]
     image_job.set_data(
         training_data=training_data,
         target_column_name=target_column_name,
@@ -111,7 +111,7 @@ def image_classification(
                 :dedent: 8
                 :caption: creating an automl image classification job
     """
-    return _create_image_job(
+    return _create_image_job(  # type: ignore[type-var, return-value]
         job_cls=ImageClassificationJob,
         training_data=training_data,
         target_column_name=target_column_name,
@@ -173,7 +173,7 @@ def image_classification_multilabel(
                 :dedent: 8
                 :caption: creating an automl image multilabel classification job
     """
-    return _create_image_job(
+    return _create_image_job(  # type: ignore[type-var, return-value]
         job_cls=ImageClassificationMultilabelJob,
         training_data=training_data,
         target_column_name=target_column_name,
@@ -234,7 +234,7 @@ def image_object_detection(
                 :dedent: 8
                 :caption: creating an automl image object detection job
     """
-    return _create_image_job(
+    return _create_image_job(  # type: ignore[type-var, return-value]
         job_cls=ImageObjectDetectionJob,
         training_data=training_data,
         target_column_name=target_column_name,
@@ -295,7 +295,7 @@ def image_instance_segmentation(
                 :dedent: 8
                 :caption: creating an automl image instance segmentation job
     """
-    return _create_image_job(
+    return _create_image_job(  # type: ignore[type-var, return-value]
         job_cls=ImageInstanceSegmentationJob,
         training_data=training_data,
         target_column_name=target_column_name,

sdk/ml/azure-ai-ml/azure/ai/ml/entities/_builders/sweep.py

@@ -176,7 +176,7 @@ def __init__(
             resources=resources,
         )
 
-        self.identity = identity
+        self.identity: Any = identity
         self._init = False
 
     @property

sdk/ml/azure-ai-ml/azure/ai/ml/entities/_job/sweep/sweep_job.py

@@ -21,7 +21,7 @@
     UserIdentityConfiguration,
     _BaseJobIdentityConfiguration,
 )
-from azure.ai.ml.entities._inputs_outputs import Input, Output
+from azure.ai.ml.entities._inputs_outputs import Input
 from azure.ai.ml.entities._job._input_output_helpers import (
     from_rest_data_outputs,
     from_rest_inputs_to_dataset_literal,
@@ -137,7 +137,7 @@ def __init__(
             Union[ManagedIdentityConfiguration, AmlTokenConfiguration, UserIdentityConfiguration]
         ] = None,
         inputs: Optional[Dict[str, Union[Input, str, bool, int, float]]] = None,
-        outputs: Optional[Dict[str, Output]] = None,
+        outputs: Optional[Dict] = None,
         compute: Optional[str] = None,
         limits: Optional[SweepJobLimits] = None,
         sampling_algorithm: Optional[Union[str, SamplingAlgorithm]] = None,

sdk/ml/azure-ai-ml/azure/ai/ml/operations/_job_operations.py

@@ -1441,12 +1441,19 @@ def _resolve_arm_id_for_sweep_job(self, job: SweepJob, resolver: _AssetResolver)
         :return: The provided SweepJob, with resolved fields
         :rtype: SweepJob
         """
-        if job.trial.code is not None and not is_ARM_id_for_resource(job.trial.code, AzureMLResourceType.CODE):
-            job.trial.code = resolver(
+        if (
+            job.trial is not None
+            and job.trial.code is not None
+            and not is_ARM_id_for_resource(job.trial.code, AzureMLResourceType.CODE)
+        ):
+            job.trial.code = resolver(  # type: ignore[assignment]
                 Code(base_path=job._base_path, path=job.trial.code),
                 azureml_type=AzureMLResourceType.CODE,
             )
-        job.trial.environment = resolver(job.trial.environment, azureml_type=AzureMLResourceType.ENVIRONMENT)
+        if job.trial is not None:
+            job.trial.environment = resolver(  # type: ignore[assignment]
+                job.trial.environment, azureml_type=AzureMLResourceType.ENVIRONMENT
+            )
         job.compute = self._resolve_compute_id(resolver, job.compute)
         return job
 

sdk/ml/azure-ai-ml/pyproject.toml

@@ -35,12 +35,9 @@ exclude = [
   "azure/ai/ml/_logging/",
   "azure/ai/ml/_telemetry/",
   "azure/ai/ml/_utils",
-  "azure/ai/ml/_internal",
-  "azure/ai/ml/automl",
   "azure/ai/ml/constants/",
   "azure/ai/ml/data_transfer/",
   "azure/ai/ml/parallel/",
-  "azure/ai/ml/sweep/",
   "azure/ai/ml/_azure_environments.py",
   "azure/ai/ml/exceptions.py",
   "azure/ai/ml/_exception_helper.py",