Fix copy tests (#11594)

Azure · May 21, 2020 · e877699 · e877699
1 parent bc01929
commit e877699
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/sdk/formrecognizer/azure-ai-formrecognizer/tests/test_copy_model.py b/sdk/formrecognizer/azure-ai-formrecognizer/tests/test_copy_model.py
@@ -21,7 +21,7 @@ class TestCopyModel(FormRecognizerTest):
     @GlobalTrainingAccountPreparer(copy=True)
     def test_copy_model_successful(self, client, container_sas_url, location, resource_id):
 
-        poller = client.begin_train_model(container_sas_url)
+        poller = client.begin_train_model(container_sas_url, use_training_labels=False)
         model = poller.result()
 
         target = client.get_copy_authorization(resource_region=location, resource_id=resource_id)
@@ -42,7 +42,7 @@ def test_copy_model_successful(self, client, container_sas_url, location, resour
     @GlobalTrainingAccountPreparer(copy=True)
     def test_copy_model_fail(self, client, container_sas_url, location, resource_id):
 
-        poller = client.begin_train_model(container_sas_url)
+        poller = client.begin_train_model(container_sas_url, use_training_labels=False)
         model = poller.result()
 
         # give an incorrect region
@@ -56,7 +56,7 @@ def test_copy_model_fail(self, client, container_sas_url, location, resource_id)
     @GlobalTrainingAccountPreparer(copy=True)
     def test_copy_model_transform(self, client, container_sas_url, location, resource_id):
 
-        poller = client.begin_train_model(container_sas_url)
+        poller = client.begin_train_model(container_sas_url, use_training_labels=False)
         model = poller.result()
 
         target = client.get_copy_authorization(resource_region=location, resource_id=resource_id)

diff --git a/sdk/formrecognizer/azure-ai-formrecognizer/tests/test_copy_model_async.py b/sdk/formrecognizer/azure-ai-formrecognizer/tests/test_copy_model_async.py
@@ -22,7 +22,7 @@ class TestCopyModelAsync(AsyncFormRecognizerTest):
     @GlobalTrainingAccountPreparer(copy=True)
     async def test_copy_model_successful(self, client, container_sas_url, location, resource_id):
 
-        model = await client.train_model(container_sas_url)
+        model = await client.train_model(container_sas_url, use_training_labels=False)
 
         target = await client.get_copy_authorization(resource_region=location, resource_id=resource_id)
 
@@ -41,7 +41,7 @@ async def test_copy_model_successful(self, client, container_sas_url, location,
     @GlobalTrainingAccountPreparer(copy=True)
     async def test_copy_model_fail(self, client, container_sas_url, location, resource_id):
 
-        model = await client.train_model(container_sas_url)
+        model = await client.train_model(container_sas_url, use_training_labels=False)
 
         # give an incorrect region
         target = await client.get_copy_authorization(resource_region="eastus", resource_id=resource_id)
@@ -53,7 +53,7 @@ async def test_copy_model_fail(self, client, container_sas_url, location, resour
     @GlobalTrainingAccountPreparer(copy=True)
     async def test_copy_model_transform(self, client, container_sas_url, location, resource_id):
 
-        model = await client.train_model(container_sas_url)
+        model = await client.train_model(container_sas_url, use_training_labels=False)
 
         target = await client.get_copy_authorization(resource_region=location, resource_id=resource_id)