diff --git a/backend/dataflow/validate_geometries.py b/backend/dataflow/validate_geometries.py index d872a622..cd897b58 100644 --- a/backend/dataflow/validate_geometries.py +++ b/backend/dataflow/validate_geometries.py @@ -100,10 +100,10 @@ def run(): with beam.Pipeline(options=pipeline_options) as p: (p - | 'Create Dataset' >> beam.Create([options.dataset.get()]) - | 'Read Data' >> beam.Map(lambda dataset: read_dataset(dataset, options.input_bucket.get())) + | 'Create Dataset' >> beam.Create([options.dataset]) + | 'Read Data' >> beam.Map(lambda dataset: read_dataset(dataset, options.input_bucket)) | 'Validate and Optimize' >> beam.ParDo(ValidateAndOptimize()) - | 'Write Results' >> beam.Map(lambda element: write_outputs(element, options.output_bucket.get())) + | 'Write Results' >> beam.Map(lambda element: write_outputs(element, options.output_bucket)) ) if __name__ == '__main__':