Skip to content

Commit

Permalink
Run conversion from main pipeline script. (#414)
Browse files Browse the repository at this point in the history
  • Loading branch information
delucchi-cmu authored Oct 18, 2024
1 parent 5717268 commit 8bc07d2
Showing 1 changed file with 4 additions and 0 deletions.
4 changes: 4 additions & 0 deletions src/hats_import/pipeline.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,11 +6,13 @@
from dask.distributed import Client

import hats_import.catalog.run_import as catalog_runner
import hats_import.hipscat_conversion.run_conversion as conversion_runner
import hats_import.index.run_index as index_runner
import hats_import.margin_cache.margin_cache as margin_runner
import hats_import.soap.run_soap as soap_runner
import hats_import.verification.run_verification as verification_runner
from hats_import.catalog.arguments import ImportArguments
from hats_import.hipscat_conversion.arguments import ConversionArguments
from hats_import.index.arguments import IndexArguments
from hats_import.margin_cache.margin_cache_arguments import MarginCacheArguments
from hats_import.runtime_arguments import RuntimeArguments
Expand Down Expand Up @@ -50,6 +52,8 @@ def pipeline_with_client(args: RuntimeArguments, client: Client):
soap_runner.run(args, client)
elif isinstance(args, VerificationArguments):
verification_runner.run(args)
elif isinstance(args, ConversionArguments):
conversion_runner.run(args, client)
else:
raise ValueError("unknown args type")

Expand Down

0 comments on commit 8bc07d2

Please sign in to comment.