Remove the explicit task key.

astronomy-commons · Nov 29, 2023 · 02ef73c · 02ef73c
1 parent 9f44b40
commit 02ef73c
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 5 deletions.
diff --git a/src/hipscat_import/catalog/run_import.py b/src/hipscat_import/catalog/run_import.py
@@ -27,7 +27,6 @@ def _map_pixels(args, client):
         futures.append(
             client.submit(
                 mr.map_to_pixels,
-                key=key,
                 input_file=file_path,
                 resume_path=args.resume_plan.tmp_path,
                 file_reader=reader_future,
@@ -53,7 +52,6 @@ def _split_pixels(args, alignment_future, client):
         futures.append(
             client.submit(
                 mr.split_pixels,
-                key=key,
                 input_file=file_path,
                 file_reader=reader_future,
                 highest_order=args.mapping_healpix_order,
@@ -85,7 +83,6 @@ def _reduce_pixels(args, destination_pixel_map, client):
         futures.append(
             client.submit(
                 mr.reduce_pixel_shards,
-                key=destination_pixel_key,
                 cache_shard_path=args.tmp_path,
                 resume_path=args.resume_plan.tmp_path,
                 reducing_key=destination_pixel_key,

diff --git a/src/hipscat_import/soap/run_soap.py b/src/hipscat_import/soap/run_soap.py
@@ -22,11 +22,10 @@ def run(args, client):
     resume_plan = SoapPlan(args)
     if not resume_plan.is_counting_done():
         futures = []
-        for source_pixel, object_pixels, source_key in resume_plan.count_keys:
+        for source_pixel, object_pixels, _source_key in resume_plan.count_keys:
             futures.append(
                 client.submit(
                     count_joins,
-                    key=source_key,
                     soap_args=args,
                     source_pixel=source_pixel,
                     object_pixels=object_pixels,