def get_generator(self) -> Iterator[BodyState]:
path = self.args.get('json_input_path')
with open(path) as f:
for line in f.readlines():
yield BodyState.from_dict(
json.loads(line)
)