apply Black 2024 style in fbcode (8/16)

Summary:
Formats the covered files with pyfmt.

paintitblack

Reviewed By: aleivag

Differential Revision: D54447737

fbshipit-source-id: 6c05d7941c6b4f1787b8da6cf810693f48a96c4e

Author: amyreese

fbpcs/bolt/oss_bolt_pcs.py

@@ -64,9 +64,11 @@ class BoltPCSCreateInstanceArgs(BoltCreateInstanceArgs, DataClassJsonMixin):
                 # if no value is provided in the yaml file, then the dataclass json
                 # library will return the default stage flow. Otherwise, if it was
                 # provided in the yaml file, we should decode the string.
-                decoder=lambda x: x
-                if x is PrivateComputationBaseStageFlow
-                else PrivateComputationBaseStageFlow.cls_name_to_cls(x),
+                decoder=lambda x: (
+                    x
+                    if x is PrivateComputationBaseStageFlow
+                    else PrivateComputationBaseStageFlow.cls_name_to_cls(x)
+                ),
             )
         },
     )

fbpcs/bolt/read_config.py

@@ -20,9 +20,7 @@
 from fbpcs.utils.config_yaml.config_yaml_dict import ConfigYamlDict
 
 
-def parse_bolt_config(
-    config: Dict[str, Any], logger: logging.Logger
-) -> Tuple[
+def parse_bolt_config(config: Dict[str, Any], logger: logging.Logger) -> Tuple[
     BoltRunner[BoltPCSCreateInstanceArgs, BoltPCSCreateInstanceArgs],
     List[BoltJob[BoltPCSCreateInstanceArgs, BoltPCSCreateInstanceArgs]],
 ]:

fbpcs/bolt/test/test_bolt_runner.py

@@ -161,7 +161,7 @@ async def test_non_joint_stage(
     @mock.patch("fbpcs.bolt.bolt_runner.asyncio.sleep")
     async def test_get_server_ips_after_start(self, mock_sleep) -> None:
         mock_server_ips = ["1.1.1.1"]
-        for (test_stage, test_publisher_status) in (
+        for test_stage, test_publisher_status in (
             (  # test Non-joint stage should get None server ips
                 DummyNonJointStageFlow.NON_JOINT_STAGE,
                 DummyNonJointStageFlow.NON_JOINT_STAGE.completed_status,

fbpcs/common/entity/dataclasses_hooks.py

@@ -31,13 +31,11 @@ def run(
         previous_field_value: Any,
         new_field_value: Any,
         hook_event: HookEventType,
-    ) -> None:
-        ...
+    ) -> None: ...
 
     @property
     @abstractmethod
-    def triggers(self) -> Iterable[HookEventType]:
-        ...
+    def triggers(self) -> Iterable[HookEventType]: ...
 
 
 @dataclass

fbpcs/common/feature/pcs_feature_gate_utils.py

@@ -37,9 +37,11 @@ def get_stage_flow(
 ) -> Type[PrivateComputationBaseStageFlow]:
     selected_stage_flow_cls = unwrap_or_default(
         optional=stage_flow_cls,
-        default=PrivateComputationPCF2StageFlow
-        if game_type is PrivateComputationGameType.ATTRIBUTION
-        else PrivateComputationStageFlow,
+        default=(
+            PrivateComputationPCF2StageFlow
+            if game_type is PrivateComputationGameType.ATTRIBUTION
+            else PrivateComputationStageFlow
+        ),
     )
 
     # warning, enabled feature gating will override stage flow, Please contact PSI team to have a similar adoption

fbpcs/common/service/test/test_pcs_container_service.py

@@ -123,14 +123,14 @@ def test_create_instances_with_list_of_env_vars(self, mock_create_instance):
         ]
 
         # Act
-        container_instances: List[
-            ContainerInstance
-        ] = self.container_svc.create_instances(
-            container_definition=f"{TEST_TASK_DEFNITION}#{TEST_CONTAINER_DEFNITION}",
-            cmds=cmd_list,
-            env_vars=[TEST_ENV_VARS, TEST_ENV_VARS_2],
-            container_type=TEST_CONTAINER_TYPE,
-            permission=TEST_CONTAINER_PERMISSION,
+        container_instances: List[ContainerInstance] = (
+            self.container_svc.create_instances(
+                container_definition=f"{TEST_TASK_DEFNITION}#{TEST_CONTAINER_DEFNITION}",
+                cmds=cmd_list,
+                env_vars=[TEST_ENV_VARS, TEST_ENV_VARS_2],
+                container_type=TEST_CONTAINER_TYPE,
+                permission=TEST_CONTAINER_PERMISSION,
+            )
         )
 
         # Assert

fbpcs/common/service/test/test_trace_logging_service.py

@@ -72,9 +72,9 @@ def test_write_checkpoint_cm(self) -> None:
                             "run_id": "run123",
                             "instance_id": "instance456",
                             "checkpoint_name": "foo",
-                            "checkpoint_data": checkpoint_data.copy()
-                            if checkpoint_data
-                            else None,
+                            "checkpoint_data": (
+                                checkpoint_data.copy() if checkpoint_data else None
+                            ),
                         }
 
                         try:

fbpcs/common/tests/entity/test_comprehensive_hooks_and_mutability.py

@@ -126,6 +126,7 @@ def region_is_deleted(obj: InstanceBase) -> bool:
 # generic hooks: track the highest pressure
 ##########################
 
+
 # keep track of the highest pressure
 def set_highest_pressure(obj: InstanceBase) -> None:
     # pyre-ignore Undefined attribute [16]: `InstanceBase` has no attribute `pressure`

fbpcs/common/tests/entity/test_generic_hook.py

@@ -13,6 +13,7 @@
 from fbpcs.common.entity.generic_hook import GenericHook
 from fbpcs.common.entity.instance_base import InstanceBase
 
+
 # keep tracking the highest pressure
 def set_highest_pressure(obj: InstanceBase) -> None:
     pressure: int = getattr(obj, "pressure", None)

fbpcs/infra/cloud_bridge/data_ingestion/data_transformation_lambda.py

@@ -204,11 +204,11 @@ def lambda_handler(
 def _parse_client_user_agent(client_user_agent: str) -> Dict[str, str]:
     parsed_fields = {}
 
-    for (regex, browserName) in BROWSER_NAME_REGEXES:
+    for regex, browserName in BROWSER_NAME_REGEXES:
         if regex.match(client_user_agent):
             parsed_fields[BROWSER_NAME] = browserName
             break
-    for (regex, deviceOs) in DEVICE_OS_REGEXES:
+    for regex, deviceOs in DEVICE_OS_REGEXES:
         if regex.match(client_user_agent):
             parsed_fields[DEVICE_OS] = deviceOs
             break