@@ -665,7 +665,12 @@ def test_start_uploading_scalars(
|
665 | 665 | self.assertEqual(mock_tracker.blob_tracker.call_count, 0)
|
666 | 666 |
|
667 | 667 | @parameterized.parameters(
|
668 |
| -{"existing_experiment": None}, {"existing_experiment": _TEST_EXPERIMENT_NAME} |
| 668 | +{"existing_experiment": None, "one_platform_run_name": None}, |
| 669 | +{"existing_experiment": None, "one_platform_run_name": "."}, |
| 670 | +{ |
| 671 | +"existing_experiment": _TEST_EXPERIMENT_NAME, |
| 672 | +"one_platform_run_name": _TEST_ONE_PLATFORM_RUN_NAME, |
| 673 | +}, |
669 | 674 | )
|
670 | 675 | @.object(
|
671 | 676 | uploader_utils.OnePlatformResourceManager,
|
@@ -686,6 +691,7 @@ def test_start_uploading_scalars_one_shot(
|
686 | 691 | experiment_tracker_mock,
|
687 | 692 | run_resource_mock,
|
688 | 693 | existing_experiment,
|
| 694 | +one_platform_run_name, |
689 | 695 | ):
|
690 | 696 | """Check that one-shot uploading stops without AbortUploadError."""
|
691 | 697 |
|
@@ -734,7 +740,7 @@ def batch_create_time_series(parent, requests):
|
734 | 740 |
|
735 | 741 | mock_rate_limiter = mock.create_autospec(uploader_utils.RateLimiter)
|
736 | 742 | mock_tracker = mock.MagicMock()
|
737 |
| -run_resource_mock.return_value = _TEST_ONE_PLATFORM_RUN_NAME |
| 743 | +run_resource_mock.return_value = one_platform_run_name |
738 | 744 | with mock..object(
|
739 | 745 | upload_tracker, "UploadTracker", return_value=mock_tracker
|
740 | 746 | ):
|
|
0 commit comments