Final big migration push [7/22] (#2394)

Summary: Pull Request resolved: #2394

Reviewed By: kinto0

Differential Revision: D54731232

fbshipit-source-id: 2cf2afd0037c164c530531a453152d3d1747eb09

fbpcs/bolt/hooks/test/test_exception_thrower.py

@@ -4,6 +4,8 @@
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-unsafe
+
 
 from unittest import IsolatedAsyncioTestCase
 from unittest.mock import Mock

fbpcs/bolt/hooks/test/test_stage_canceller.py

@@ -4,6 +4,8 @@
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-unsafe
+
 
 from unittest import IsolatedAsyncioTestCase
 from unittest.mock import AsyncMock

fbpcs/bolt/test/test_bolt_hook.py

@@ -4,6 +4,8 @@
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-unsafe
+
 from dataclasses import dataclass
 from typing import List, TypeVar
 from unittest import IsolatedAsyncioTestCase

fbpcs/bolt/test/test_bolt_job.py

@@ -4,6 +4,8 @@
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-unsafe
+
 import unittest
 from unittest import mock
 

fbpcs/bolt/test/test_bolt_runner.py

@@ -4,6 +4,8 @@
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-unsafe
+
 import time
 import unittest
 from enum import Enum

fbpcs/bolt/test/test_oss_bolt_pcs.py

@@ -4,6 +4,8 @@
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-unsafe
+
 
 import unittest
 from collections import defaultdict

fbpcs/common/entity/dataclasses_hooks.py

@@ -3,6 +3,8 @@
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-unsafe
+
 
 from abc import abstractmethod
 from dataclasses import dataclass

fbpcs/common/entity/dataclasses_mutability.py

@@ -3,6 +3,8 @@
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-unsafe
+
 
 from dataclasses import dataclass, field
 from enum import Enum

fbpcs/common/entity/exceptions.py

@@ -3,6 +3,8 @@
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-unsafe
+
 from typing import Any
 
 

fbpcs/common/entity/update_generic_hook.py

@@ -3,6 +3,8 @@
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-unsafe
+
 
 from typing import Any, Callable, Iterable, Optional, TypeVar
 

fbpcs/common/entity/update_other_field_hook.py

@@ -3,6 +3,8 @@
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-unsafe
+
 
 from typing import Any, Callable, Iterable, Optional, TypeVar
 

fbpcs/common/service/input_data_service.py

@@ -3,6 +3,8 @@
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-unsafe
+
 import logging
 import re
 from dataclasses import dataclass

fbpcs/common/service/test/test_input_data_service.py

@@ -4,6 +4,8 @@
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-unsafe
+
 import unittest
 from unittest.mock import patch
 

fbpcs/common/service/test/test_pcs_container_service.py

@@ -4,6 +4,8 @@
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-unsafe
+
 import unittest
 from typing import List, Optional
 from unittest.mock import call, MagicMock, patch

fbpcs/common/service/test/test_pii_scrubber.py

@@ -4,6 +4,8 @@
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-unsafe
+
 from unittest import TestCase
 
 from fbpcs.common.service.pii_scrubber import PiiLoggingScrubber

fbpcs/common/service/test/test_secret_scrubber.py

@@ -4,6 +4,8 @@
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-unsafe
+
 from unittest import TestCase
 
 from fbpcs.common.service.secret_scrubber import SecretScrubber

fbpcs/common/service/test/test_trace_logging_service.py

@@ -4,6 +4,8 @@
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-unsafe
+
 from typing import Dict
 from unittest import TestCase
 

fbpcs/common/tests/entity/test_comprehensive_hooks_and_mutability.py

@@ -3,6 +3,8 @@
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-unsafe
+
 # pyer-strict
 
 import unittest

fbpcs/common/tests/entity/test_frozen_field_hook.py

@@ -3,6 +3,8 @@
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-unsafe
+
 # pyer-strict
 
 import unittest

fbpcs/common/tests/entity/test_instance_base_mutability.py

@@ -3,6 +3,8 @@
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-unsafe
+
 import unittest
 from dataclasses import dataclass
 from typing import Generic, List, Optional, TypeVar

fbpcs/common/tests/entity/test_range_hook.py

@@ -3,6 +3,8 @@
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-unsafe
+
 # pyer-strict
 
 import unittest

fbpcs/common/tests/entity/test_update_other_field_hook.py

@@ -3,6 +3,8 @@
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-unsafe
+
 # pyer-strict
 
 import unittest

fbpcs/common/tests/repository/test_instance_local.py

@@ -4,6 +4,8 @@
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-unsafe
+
 import copy
 import unittest
 from pathlib import Path

fbpcs/common/tests/test_stage_state_instance.py

@@ -4,6 +4,8 @@
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-unsafe
+
 import dataclasses
 import unittest
 from unittest.mock import MagicMock, patch

fbpcs/data_processing/pid_preparer/preparer.py

@@ -4,6 +4,8 @@
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-unsafe
+
 import abc
 import logging
 import pathlib

fbpcs/data_processing/service/id_spine_combiner.py

@@ -4,6 +4,8 @@
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-unsafe
+
 from typing import List, Optional
 
 from fbpcp.util.arg_builder import build_cmd_args

fbpcs/data_processing/service/pid_prepare_binary_service.py

@@ -4,6 +4,8 @@
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-unsafe
+
 
 from typing import Optional
 

fbpcs/data_processing/service/pid_run_protocol_binary_service.py

@@ -4,6 +4,8 @@
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-unsafe
+
 from dataclasses import dataclass
 from typing import Optional
 

fbpcs/experimental/cloud_logs/log_retriever.py

@@ -4,6 +4,8 @@
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-unsafe
+
 from abc import ABC, abstractmethod
 from typing import List
 

fbpcs/experimental/cloud_logs/test/test_log_retriever.py

@@ -4,6 +4,8 @@
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-unsafe
+
 import unittest
 
 from fbpcp.service.log_cloudwatch import CloudWatchLogService

fbpcs/infra/cloud_bridge/data_ingestion/tests/data_transformation_lambda_test.py

@@ -3,6 +3,8 @@
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-unsafe
+
 import base64
 import json
 from datetime import datetime

fbpcs/infra/logging_service/download_logs/cloud/cloud_baseclass.py

@@ -3,6 +3,8 @@
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-unsafe
+
 from abc import ABC
 
 

fbpcs/infra/restore_run_state/tests/test_restore_state.py

@@ -4,6 +4,8 @@
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-unsafe
+
 
 from unittest import mock, TestCase
 from unittest.mock import ANY

fbpcs/pc_pre_validation/tests/binary_path_test.py

@@ -3,6 +3,8 @@
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-unsafe
+
 from unittest import TestCase
 
 from fbpcs.pc_pre_validation.binary_path import (

fbpcs/pl_coordinator/bolt_graphapi_client.py

@@ -4,6 +4,8 @@
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-unsafe
+
 import json
 import logging
 import os

fbpcs/pl_coordinator/pl_study_runner.py

@@ -4,6 +4,8 @@
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-unsafe
+
 
 import asyncio
 import calendar

fbpcs/pl_coordinator/tests/test_bolt_graphapi_client.py

@@ -4,6 +4,8 @@
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-unsafe
+
 import json
 import unittest
 from typing import Any

fbpcs/pl_coordinator/tests/test_token_validator.py

@@ -4,6 +4,8 @@
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-unsafe
+
 
 import json
 from datetime import datetime, timedelta

fbpcs/private_computation/entity/exceptions.py

@@ -3,6 +3,8 @@
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-unsafe
+
 
 class PCInfraConfigError(Exception):
     pass

fbpcs/private_computation/entity/infra_config.py

@@ -3,6 +3,8 @@
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-unsafe
+
 import time
 from dataclasses import dataclass, field
 from enum import Enum

fbpcs/private_computation/entity/pc_infra_config_data.py

@@ -3,6 +3,8 @@
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-unsafe
+
 from dataclasses import dataclass
 from enum import Enum
 from typing import Set

fbpcs/private_computation/entity/product_config.py

@@ -3,6 +3,8 @@
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-unsafe
+
 from dataclasses import dataclass, field
 from enum import Enum, IntEnum
 from typing import Any, Dict, Optional

fbpcs/private_computation/pc_attribution_runner.py

@@ -4,6 +4,8 @@
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-unsafe
+
 
 import asyncio
 

fbpcs/private_computation/service/errors.py

@@ -4,6 +4,8 @@
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-unsafe
+
 
 class PrivateComputationServiceBaseException(Exception):
     pass