diff --git a/sdk/healthdataaiservices/azure-health-deidentification/azure/health/deidentification/_model_base.py b/sdk/healthdataaiservices/azure-health-deidentification/azure/health/deidentification/_model_base.py index 095e62400e89..ae6d860406b0 100644 --- a/sdk/healthdataaiservices/azure-health-deidentification/azure/health/deidentification/_model_base.py +++ b/sdk/healthdataaiservices/azure-health-deidentification/azure/health/deidentification/_model_base.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines,line-too-long,useless-suppression +# pylint: disable=too-many-lines,line-too-long,useless-suppression,unidiomatic-typecheck # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. diff --git a/sdk/healthdataaiservices/azure-health-deidentification/azure/health/deidentification/_utils/model_base.py b/sdk/healthdataaiservices/azure-health-deidentification/azure/health/deidentification/_utils/model_base.py index aaa6692b2346..3306c5314897 100644 --- a/sdk/healthdataaiservices/azure-health-deidentification/azure/health/deidentification/_utils/model_base.py +++ b/sdk/healthdataaiservices/azure-health-deidentification/azure/health/deidentification/_utils/model_base.py @@ -1,4 +1,4 @@ -# pylint: disable=line-too-long,useless-suppression,too-many-lines +# pylint: disable=line-too-long,useless-suppression,too-many-lines,unidiomatic-typecheck # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. diff --git a/sdk/healthdataaiservices/azure-health-deidentification/samples/async_samples/deidentify_documents_async.py b/sdk/healthdataaiservices/azure-health-deidentification/samples/async_samples/deidentify_documents_async.py index fe065600659e..4da287ace90a 100644 --- a/sdk/healthdataaiservices/azure-health-deidentification/samples/async_samples/deidentify_documents_async.py +++ b/sdk/healthdataaiservices/azure-health-deidentification/samples/async_samples/deidentify_documents_async.py @@ -23,6 +23,8 @@ import asyncio +import os +import uuid from azure.core.polling import AsyncLROPoller from azure.health.deidentification.aio import DeidentificationClient from azure.health.deidentification.models import ( @@ -31,8 +33,6 @@ TargetStorageLocation, ) from azure.identity.aio import DefaultAzureCredential -import os -import uuid async def deidentify_documents_async(): diff --git a/sdk/healthdataaiservices/azure-health-deidentification/samples/async_samples/deidentify_text_redact_async.py b/sdk/healthdataaiservices/azure-health-deidentification/samples/async_samples/deidentify_text_redact_async.py index 2563e8d8c69e..f33ba9eda6f7 100644 --- a/sdk/healthdataaiservices/azure-health-deidentification/samples/async_samples/deidentify_text_redact_async.py +++ b/sdk/healthdataaiservices/azure-health-deidentification/samples/async_samples/deidentify_text_redact_async.py @@ -19,6 +19,7 @@ import asyncio +import os from azure.health.deidentification.aio import DeidentificationClient from azure.health.deidentification.models import ( DeidentificationContent, @@ -26,7 +27,6 @@ DeidentificationResult, ) from azure.identity.aio import DefaultAzureCredential -import os async def deidentify_text_redact_async(): diff --git a/sdk/healthdataaiservices/azure-health-deidentification/samples/async_samples/deidentify_text_surrogate_async.py b/sdk/healthdataaiservices/azure-health-deidentification/samples/async_samples/deidentify_text_surrogate_async.py index aefa7eea4f63..36226b07e0f2 100644 --- a/sdk/healthdataaiservices/azure-health-deidentification/samples/async_samples/deidentify_text_surrogate_async.py +++ b/sdk/healthdataaiservices/azure-health-deidentification/samples/async_samples/deidentify_text_surrogate_async.py @@ -19,14 +19,13 @@ import asyncio +import os from azure.health.deidentification.aio import DeidentificationClient from azure.health.deidentification.models import ( DeidentificationContent, - DeidentificationOperationType, DeidentificationResult, ) from azure.identity.aio import DefaultAzureCredential -import os async def deidentify_text_surrogate_async(): diff --git a/sdk/healthdataaiservices/azure-health-deidentification/samples/async_samples/deidentify_text_surrogate_only_async.py b/sdk/healthdataaiservices/azure-health-deidentification/samples/async_samples/deidentify_text_surrogate_only_async.py index 62008166d747..f094f0b12d1a 100644 --- a/sdk/healthdataaiservices/azure-health-deidentification/samples/async_samples/deidentify_text_surrogate_only_async.py +++ b/sdk/healthdataaiservices/azure-health-deidentification/samples/async_samples/deidentify_text_surrogate_only_async.py @@ -19,6 +19,7 @@ import asyncio +import os from azure.health.deidentification.aio import DeidentificationClient from azure.health.deidentification.models import ( DeidentificationContent, @@ -31,7 +32,6 @@ TextEncodingType, ) from azure.identity.aio import DefaultAzureCredential -import os async def deidentify_text_surrogate_only_async(): diff --git a/sdk/healthdataaiservices/azure-health-deidentification/samples/async_samples/deidentify_text_tag_async.py b/sdk/healthdataaiservices/azure-health-deidentification/samples/async_samples/deidentify_text_tag_async.py index 49b19ae181e4..a31e6292d943 100644 --- a/sdk/healthdataaiservices/azure-health-deidentification/samples/async_samples/deidentify_text_tag_async.py +++ b/sdk/healthdataaiservices/azure-health-deidentification/samples/async_samples/deidentify_text_tag_async.py @@ -19,6 +19,7 @@ import asyncio +import os from azure.health.deidentification.aio import DeidentificationClient from azure.health.deidentification.models import ( DeidentificationContent, @@ -26,7 +27,6 @@ DeidentificationResult, ) from azure.identity.aio import DefaultAzureCredential -import os async def deidentify_text_tag_async(): @@ -43,7 +43,7 @@ async def deidentify_text_tag_async(): print(f'\nOriginal Text: "{body.input_text}"') if result.tagger_result and result.tagger_result.entities: - print(f"Tagged Entities:") + print("Tagged Entities:") for entity in result.tagger_result.entities: print( f'\tEntity Text: "{entity.text}", Entity Category: "{entity.category}", Offset: "{entity.offset.code_point}", Length: "{entity.length.code_point}"' diff --git a/sdk/healthdataaiservices/azure-health-deidentification/samples/async_samples/list_job_documents_async.py b/sdk/healthdataaiservices/azure-health-deidentification/samples/async_samples/list_job_documents_async.py index 54ea3ec0464c..2025a815a5bc 100644 --- a/sdk/healthdataaiservices/azure-health-deidentification/samples/async_samples/list_job_documents_async.py +++ b/sdk/healthdataaiservices/azure-health-deidentification/samples/async_samples/list_job_documents_async.py @@ -22,6 +22,8 @@ import asyncio +import os +import uuid from azure.core.polling import AsyncLROPoller from azure.health.deidentification.aio import DeidentificationClient from azure.health.deidentification.models import ( @@ -30,8 +32,6 @@ TargetStorageLocation, ) from azure.identity.aio import DefaultAzureCredential -import os -import uuid async def list_job_documents_async(): diff --git a/sdk/healthdataaiservices/azure-health-deidentification/samples/async_samples/list_jobs_async.py b/sdk/healthdataaiservices/azure-health-deidentification/samples/async_samples/list_jobs_async.py index caed2ca594e5..9febaa3ed0d0 100644 --- a/sdk/healthdataaiservices/azure-health-deidentification/samples/async_samples/list_jobs_async.py +++ b/sdk/healthdataaiservices/azure-health-deidentification/samples/async_samples/list_jobs_async.py @@ -18,9 +18,9 @@ import asyncio +import os from azure.health.deidentification.aio import DeidentificationClient from azure.identity.aio import DefaultAzureCredential -import os async def list_jobs_async(): diff --git a/sdk/healthdataaiservices/azure-health-deidentification/samples/deidentify_text_tag.py b/sdk/healthdataaiservices/azure-health-deidentification/samples/deidentify_text_tag.py index ee6725e260fe..819c4fc0e05a 100644 --- a/sdk/healthdataaiservices/azure-health-deidentification/samples/deidentify_text_tag.py +++ b/sdk/healthdataaiservices/azure-health-deidentification/samples/deidentify_text_tag.py @@ -40,7 +40,7 @@ def deidentify_text_tag(): print(f'\nOriginal Text: "{body.input_text}"') if result.tagger_result and result.tagger_result.entities: - print(f"Tagged Entities:") + print("Tagged Entities:") for entity in result.tagger_result.entities: print( f'\tEntity Text: "{entity.text}", Entity Category: "{entity.category}", Offset: "{entity.offset.code_point}", Length: "{entity.length.code_point}"' diff --git a/sdk/healthdataaiservices/azure-health-deidentification/samples/examples.py b/sdk/healthdataaiservices/azure-health-deidentification/samples/examples.py index 5bf55a218ee9..f23dd8092875 100644 --- a/sdk/healthdataaiservices/azure-health-deidentification/samples/examples.py +++ b/sdk/healthdataaiservices/azure-health-deidentification/samples/examples.py @@ -1,21 +1,22 @@ -# [START create_client] +# ------------------------------------ +# Copyright (c) Microsoft Corporation. +# Licensed under the MIT License. +# ------------------------------------ +import os +from azure.core.exceptions import AzureError from azure.health.deidentification import DeidentificationClient +from azure.health.deidentification.models import ( + DeidentificationContent, +) from azure.identity import DefaultAzureCredential -import os - +# [START create_client] endpoint = os.environ["HEALTHDATAAISERVICES_DEID_SERVICE_ENDPOINT"] credential = DefaultAzureCredential() client = DeidentificationClient(endpoint, credential) # [END create_client] # [START handle_error] -from azure.core.exceptions import AzureError -from azure.health.deidentification.models import ( - DeidentificationContent, -) - - error_client = DeidentificationClient("https://contoso.deid.azure.com", credential) body = DeidentificationContent(input_text="Hello, I'm Dr. John Smith.") diff --git a/sdk/healthdataaiservices/azure-health-deidentification/samples/list_job_documents.py b/sdk/healthdataaiservices/azure-health-deidentification/samples/list_job_documents.py index 1322e0a75285..a47e995ef317 100644 --- a/sdk/healthdataaiservices/azure-health-deidentification/samples/list_job_documents.py +++ b/sdk/healthdataaiservices/azure-health-deidentification/samples/list_job_documents.py @@ -23,6 +23,8 @@ """ +import os +import uuid from azure.health.deidentification import DeidentificationClient from azure.health.deidentification.models import ( DeidentificationJob, @@ -30,8 +32,6 @@ TargetStorageLocation, ) from azure.identity import DefaultAzureCredential -import os -import uuid def list_job_documents(): diff --git a/sdk/healthdataaiservices/azure-health-deidentification/samples/list_jobs.py b/sdk/healthdataaiservices/azure-health-deidentification/samples/list_jobs.py index 2de4081b8139..248b9a0107dd 100644 --- a/sdk/healthdataaiservices/azure-health-deidentification/samples/list_jobs.py +++ b/sdk/healthdataaiservices/azure-health-deidentification/samples/list_jobs.py @@ -17,9 +17,9 @@ """ +import os from azure.health.deidentification import DeidentificationClient from azure.identity import DefaultAzureCredential -import os def list_jobs(): diff --git a/sdk/healthdataaiservices/azure-health-deidentification/tests/deid_base_test_case.py b/sdk/healthdataaiservices/azure-health-deidentification/tests/deid_base_test_case.py index cb9bb597f468..aeef540390b8 100644 --- a/sdk/healthdataaiservices/azure-health-deidentification/tests/deid_base_test_case.py +++ b/sdk/healthdataaiservices/azure-health-deidentification/tests/deid_base_test_case.py @@ -1,19 +1,15 @@ import functools import inspect import os -import random -import datetime -import uuid -from azure.health.deidentification import DeidentificationClient -from azure.health.deidentification.aio import ( - DeidentificationClient as DeidentificationClientAsync, -) - from devtools_testutils import ( AzureRecordedTestCase, EnvironmentVariableLoader, ) +from azure.health.deidentification import DeidentificationClient +from azure.health.deidentification.aio import ( + DeidentificationClient as DeidentificationClientAsync, +) RealtimeEnv = functools.partial( EnvironmentVariableLoader, diff --git a/sdk/healthdataaiservices/azure-health-deidentification/tests/test_create_delete.py b/sdk/healthdataaiservices/azure-health-deidentification/tests/test_create_delete.py index 9274dab6f955..5be2123ba8ba 100644 --- a/sdk/healthdataaiservices/azure-health-deidentification/tests/test_create_delete.py +++ b/sdk/healthdataaiservices/azure-health-deidentification/tests/test_create_delete.py @@ -1,13 +1,16 @@ -from datetime import time -from azure.core.exceptions import ResourceNotFoundError - import pytest from deid_base_test_case import DeidBaseTestCase, BatchEnv from devtools_testutils import ( recorded_by_proxy, ) - -from azure.health.deidentification.models import * +from azure.core.exceptions import ResourceNotFoundError +from azure.health.deidentification.models import ( + DeidentificationJob, + DeidentificationOperationType, + OperationStatus, + SourceStorageLocation, + TargetStorageLocation, +) class TestHealthDeidentificationCreateCancelDelete(DeidBaseTestCase): diff --git a/sdk/healthdataaiservices/azure-health-deidentification/tests/test_create_delete_async.py b/sdk/healthdataaiservices/azure-health-deidentification/tests/test_create_delete_async.py index 630b005b8c69..6b2e4509b7f0 100644 --- a/sdk/healthdataaiservices/azure-health-deidentification/tests/test_create_delete_async.py +++ b/sdk/healthdataaiservices/azure-health-deidentification/tests/test_create_delete_async.py @@ -1,13 +1,15 @@ -from datetime import time -from azure.core.exceptions import ResourceNotFoundError - import pytest from deid_base_test_case import DeidBaseTestCase, BatchEnv from devtools_testutils.aio import ( recorded_by_proxy_async, ) - -from azure.health.deidentification.models import * +from azure.core.exceptions import ResourceNotFoundError +from azure.health.deidentification.models import ( + DeidentificationJob, + OperationStatus, + SourceStorageLocation, + TargetStorageLocation, +) class TestHealthDeidentificationCreateCancelDelete(DeidBaseTestCase): diff --git a/sdk/healthdataaiservices/azure-health-deidentification/tests/test_create_list.py b/sdk/healthdataaiservices/azure-health-deidentification/tests/test_create_list.py index fa76b27d3b22..66d6b3c3ad32 100644 --- a/sdk/healthdataaiservices/azure-health-deidentification/tests/test_create_list.py +++ b/sdk/healthdataaiservices/azure-health-deidentification/tests/test_create_list.py @@ -1,10 +1,16 @@ -from deid_base_test_case import * +from deid_base_test_case import DeidBaseTestCase, BatchEnv from devtools_testutils import ( recorded_by_proxy, ) -from azure.health.deidentification.models import * -from azure.core.polling import LROPoller +from azure.health.deidentification.models import ( + DeidentificationJob, + DeidentificationJobCustomizationOptions, + DeidentificationOperationType, + OperationStatus, + SourceStorageLocation, + TargetStorageLocation, +) class TestHealthDeidentificationCreateAndListJob(DeidBaseTestCase): @@ -39,12 +45,12 @@ def test_create_list(self, **kwargs): if j.job_name == jobname: job = j break - elif jobsToLookThrough <= 0: - raise Exception("Job not found in list_jobs") + if jobsToLookThrough <= 0: + raise RuntimeError("Job not found in list_jobs") assert job is not None assert job.job_name == jobname - assert job.status == OperationStatus.NOT_STARTED or job.status == OperationStatus.RUNNING + assert job.status in (OperationStatus.NOT_STARTED, OperationStatus.RUNNING) assert job.operation_type == DeidentificationOperationType.REDACT assert job.error is None assert job.created_at is not None diff --git a/sdk/healthdataaiservices/azure-health-deidentification/tests/test_create_list_async.py b/sdk/healthdataaiservices/azure-health-deidentification/tests/test_create_list_async.py index 161a8665e7e1..e2ebfd04c823 100644 --- a/sdk/healthdataaiservices/azure-health-deidentification/tests/test_create_list_async.py +++ b/sdk/healthdataaiservices/azure-health-deidentification/tests/test_create_list_async.py @@ -1,10 +1,17 @@ -from deid_base_test_case import * +import pytest +from deid_base_test_case import DeidBaseTestCase, BatchEnv from devtools_testutils.aio import ( recorded_by_proxy_async, ) -from azure.health.deidentification.models import * -import pytest +from azure.health.deidentification.models import ( + DeidentificationJob, + DeidentificationJobCustomizationOptions, + DeidentificationOperationType, + OperationStatus, + SourceStorageLocation, + TargetStorageLocation, +) class TestHealthDeidentificationCreateAndListJob(DeidBaseTestCase): @@ -40,12 +47,12 @@ async def test_create_list_async(self, **kwargs): if j.job_name == jobname: job = j break - elif jobsToLookThrough <= 0: - raise Exception("Job not found in list_jobs") + if jobsToLookThrough <= 0: + raise RuntimeError("Job not found in list_jobs") assert job is not None assert job.job_name == jobname - assert job.status == OperationStatus.NOT_STARTED or job.status == OperationStatus.RUNNING + assert job.status in (OperationStatus.NOT_STARTED, OperationStatus.RUNNING) assert job.operation_type == DeidentificationOperationType.REDACT assert job.error is None assert job.created_at is not None diff --git a/sdk/healthdataaiservices/azure-health-deidentification/tests/test_create_wait_finish.py b/sdk/healthdataaiservices/azure-health-deidentification/tests/test_create_wait_finish.py index aaebe91349a6..3572001e38e6 100644 --- a/sdk/healthdataaiservices/azure-health-deidentification/tests/test_create_wait_finish.py +++ b/sdk/healthdataaiservices/azure-health-deidentification/tests/test_create_wait_finish.py @@ -1,10 +1,16 @@ -from deid_base_test_case import * +from deid_base_test_case import DeidBaseTestCase, BatchEnv from devtools_testutils import ( recorded_by_proxy, ) -from azure.health.deidentification.models import * from azure.core.polling import LROPoller +from azure.health.deidentification.models import ( + DeidentificationJob, + DeidentificationOperationType, + OperationStatus, + SourceStorageLocation, + TargetStorageLocation, +) class TestHealthDeidentificationCreateJobWaitUntil(DeidBaseTestCase): diff --git a/sdk/healthdataaiservices/azure-health-deidentification/tests/test_create_wait_finish_async.py b/sdk/healthdataaiservices/azure-health-deidentification/tests/test_create_wait_finish_async.py index be9bae5499fc..455d0b14a03f 100644 --- a/sdk/healthdataaiservices/azure-health-deidentification/tests/test_create_wait_finish_async.py +++ b/sdk/healthdataaiservices/azure-health-deidentification/tests/test_create_wait_finish_async.py @@ -1,11 +1,17 @@ -from deid_base_test_case import * +import pytest +from deid_base_test_case import DeidBaseTestCase, BatchEnv from devtools_testutils.aio import ( recorded_by_proxy_async, ) -from azure.health.deidentification.models import * from azure.core.polling import AsyncLROPoller -import pytest +from azure.health.deidentification.models import ( + DeidentificationJob, + DeidentificationOperationType, + OperationStatus, + SourceStorageLocation, + TargetStorageLocation, +) class TestHealthDeidentificationCreateJobWaitUntil(DeidBaseTestCase): diff --git a/sdk/healthdataaiservices/azure-health-deidentification/tests/test_exception_throws.py b/sdk/healthdataaiservices/azure-health-deidentification/tests/test_exception_throws.py index 328db5d2777c..1e87e39e0b29 100644 --- a/sdk/healthdataaiservices/azure-health-deidentification/tests/test_exception_throws.py +++ b/sdk/healthdataaiservices/azure-health-deidentification/tests/test_exception_throws.py @@ -1,15 +1,17 @@ -from datetime import time -from time import sleep -from azure.core.exceptions import HttpResponseError - import pytest from deid_base_test_case import DeidBaseTestCase, BatchEnv from devtools_testutils import ( recorded_by_proxy, ) - -from azure.health.deidentification.models import * +from azure.core.exceptions import HttpResponseError from azure.core.polling import LROPoller +from azure.health.deidentification.models import ( + DeidentificationJob, + DeidentificationOperationType, + OperationStatus, + SourceStorageLocation, + TargetStorageLocation, +) class TestHealthDeidentificationExceptionThrows(DeidBaseTestCase): diff --git a/sdk/healthdataaiservices/azure-health-deidentification/tests/test_exception_throws_async.py b/sdk/healthdataaiservices/azure-health-deidentification/tests/test_exception_throws_async.py index 5e5bb47cd7ef..816506890d99 100644 --- a/sdk/healthdataaiservices/azure-health-deidentification/tests/test_exception_throws_async.py +++ b/sdk/healthdataaiservices/azure-health-deidentification/tests/test_exception_throws_async.py @@ -1,15 +1,17 @@ -from datetime import time -from time import sleep -from azure.core.exceptions import HttpResponseError - import pytest from deid_base_test_case import DeidBaseTestCase, BatchEnv from devtools_testutils.aio import ( recorded_by_proxy_async, ) - -from azure.health.deidentification.models import * +from azure.core.exceptions import HttpResponseError from azure.core.polling import AsyncLROPoller +from azure.health.deidentification.models import ( + DeidentificationJob, + DeidentificationOperationType, + OperationStatus, + SourceStorageLocation, + TargetStorageLocation, +) class TestHealthDeidentificationExceptionThrows(DeidBaseTestCase): diff --git a/sdk/healthdataaiservices/azure-health-deidentification/tests/test_hello_world.py b/sdk/healthdataaiservices/azure-health-deidentification/tests/test_hello_world.py index 1f1e291d8496..fe43e5d0cabe 100644 --- a/sdk/healthdataaiservices/azure-health-deidentification/tests/test_hello_world.py +++ b/sdk/healthdataaiservices/azure-health-deidentification/tests/test_hello_world.py @@ -3,7 +3,11 @@ recorded_by_proxy, ) -from azure.health.deidentification.models import * +from azure.health.deidentification.models import ( + DeidentificationContent, + DeidentificationOperationType, + DeidentificationResult, +) class TestHealthDeidentificationHelloWorld(DeidBaseTestCase): diff --git a/sdk/healthdataaiservices/azure-health-deidentification/tests/test_hello_world_async.py b/sdk/healthdataaiservices/azure-health-deidentification/tests/test_hello_world_async.py index 771b694b4864..b19ccab2e3ca 100644 --- a/sdk/healthdataaiservices/azure-health-deidentification/tests/test_hello_world_async.py +++ b/sdk/healthdataaiservices/azure-health-deidentification/tests/test_hello_world_async.py @@ -1,10 +1,14 @@ +import pytest from deid_base_test_case import DeidBaseTestCase, RealtimeEnv from devtools_testutils.aio import ( recorded_by_proxy_async, ) -import pytest -from azure.health.deidentification.models import * +from azure.health.deidentification.models import ( + DeidentificationContent, + DeidentificationOperationType, + DeidentificationResult, +) class TestHealthDeidentificationHelloWorld(DeidBaseTestCase): diff --git a/sdk/healthdataaiservices/azure-health-deidentification/tests/test_list_job_docs_pagination.py b/sdk/healthdataaiservices/azure-health-deidentification/tests/test_list_job_docs_pagination.py index ddb7f06ca83f..21a5081166d3 100644 --- a/sdk/healthdataaiservices/azure-health-deidentification/tests/test_list_job_docs_pagination.py +++ b/sdk/healthdataaiservices/azure-health-deidentification/tests/test_list_job_docs_pagination.py @@ -1,10 +1,16 @@ -from deid_base_test_case import * +from deid_base_test_case import DeidBaseTestCase, BatchEnv from devtools_testutils import ( recorded_by_proxy, ) -from azure.health.deidentification.models import * from azure.core.paging import ItemPaged +from azure.health.deidentification.models import ( + DeidentificationJob, + DeidentificationJobCustomizationOptions, + DeidentificationOperationType, + SourceStorageLocation, + TargetStorageLocation, +) class TestHealthDeidentificationCreateAndListJob(DeidBaseTestCase): diff --git a/sdk/healthdataaiservices/azure-health-deidentification/tests/test_surrogate_only.py b/sdk/healthdataaiservices/azure-health-deidentification/tests/test_surrogate_only.py index 7dfca93f4199..35bd6edb81e5 100644 --- a/sdk/healthdataaiservices/azure-health-deidentification/tests/test_surrogate_only.py +++ b/sdk/healthdataaiservices/azure-health-deidentification/tests/test_surrogate_only.py @@ -3,7 +3,15 @@ recorded_by_proxy, ) -from azure.health.deidentification.models import * +from azure.health.deidentification.models import ( + DeidentificationContent, + DeidentificationOperationType, + DeidentificationResult, + PhiCategory, + SimplePhiEntity, + TaggedPhiEntities, + TextEncodingType, +) class TestHealthDeidentificationHelloWorld(DeidBaseTestCase): diff --git a/sdk/healthdataaiservices/azure-health-deidentification/tests/test_surrogate_only_async.py b/sdk/healthdataaiservices/azure-health-deidentification/tests/test_surrogate_only_async.py index 44b2a725d0c5..12b6dd02ef0c 100644 --- a/sdk/healthdataaiservices/azure-health-deidentification/tests/test_surrogate_only_async.py +++ b/sdk/healthdataaiservices/azure-health-deidentification/tests/test_surrogate_only_async.py @@ -4,7 +4,15 @@ ) import pytest -from azure.health.deidentification.models import * +from azure.health.deidentification.models import ( + DeidentificationContent, + DeidentificationOperationType, + DeidentificationResult, + PhiCategory, + SimplePhiEntity, + TaggedPhiEntities, + TextEncodingType, +) class TestHealthDeidentificationHelloWorld(DeidBaseTestCase):