diff --git a/pyproject.toml b/pyproject.toml index 828c583..9c86622 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -24,6 +24,7 @@ dependencies = [ "pz-rail-pipelines", "matplotlib", "yamllint", + "pyarrow", ] # On a mac, install optional dependencies with `pip install '.[dev]'` (include the single quotes) [project.optional-dependencies] diff --git a/src/rail/cli/rail_project/project_commands.py b/src/rail/cli/rail_project/project_commands.py index a69e46d..642ef56 100644 --- a/src/rail/cli/rail_project/project_commands.py +++ b/src/rail/cli/rail_project/project_commands.py @@ -59,10 +59,11 @@ def subsample_command(config_file: str, **kwargs: Any) -> int: @project_options.run_mode() @project_options.site() @project_options.args() -def sbatch_command(run_mode: project_options.RunMode, site: str, args: list[str]) -> int: +def sbatch_command(run_mode: project_options.RunMode, site: str, args: list[str]) -> int: # pragma: no cover """Wrap a rail_pipe command with site-based arguements for slurm""" return project_scripts.sbatch_wrap(run_mode, site, args) + @project_cli.group(name="reduce") def reduce_group() -> None: """Reduce input data for PZ analysis"""