Skip to content

Commit

Permalink
Remove fieldvalues in test data
Browse files Browse the repository at this point in the history
  • Loading branch information
jacob720 committed Jan 9, 2025
1 parent 09f19f4 commit 5382205
Show file tree
Hide file tree
Showing 3 changed files with 10,027 additions and 0 deletions.
25 changes: 25 additions & 0 deletions tests/remove_field_values_for_testing.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,25 @@
# Script to remove field values to make data used for testing cleaner

import argparse
from pathlib import Path

from aa_remove_data.pb_utils import PBUtils


def get_args():
parser = argparse.ArgumentParser()
parser.add_argument("filename")
return parser.parse_args()


def remove_field_values(filename):
pb = PBUtils(filename)
for sample in pb.samples:
sample.ClearField("fieldvalues")
pb.write_pb(filename)


if __name__ == "__main__":
args = get_args()
filename = Path(args.filename)
remove_field_values(filename)
Binary file modified tests/test_data/P:2021_short.pb
Binary file not shown.
Loading

0 comments on commit 5382205

Please sign in to comment.