Shard our taint output files for the distillery integration test

Summary: The taint-output.json file is getting quite large (2GB), let's shard it.

Reviewed By: tianhan0

Differential Revision: D77724865

fbshipit-source-id: 63072e2c0a51c117991ef352bf9f0e52ebe0930e

Author: arthaud

tools/pysa_integration_tests/runner_lib.py

@@ -218,6 +218,7 @@ def run_pysa(
     passthrough_args: Optional[Sequence[str]] = None,
     working_directory: Optional[Path] = None,
     silent: bool = False,
+    shard_taint_output: bool = False,
     error_help: Optional[str] = None,
 ) -> str:
     """Run pysa for the given test and produce a list of errors in JSON."""
@@ -270,6 +271,9 @@ def run_pysa(
     if maximum_tito_depth is not None:
         command.append(f"--maximum-tito-depth={maximum_tito_depth }")
 
+    if shard_taint_output:
+        command.append("--output-format=sharded-json")
+
     if passthrough_args is not None:
         command.extend(passthrough_args)