Skip to content

Commit

Permalink
Better mypy hints (#316)
Browse files Browse the repository at this point in the history
  • Loading branch information
delucchi-cmu authored Jul 26, 2024
1 parent 3234fcf commit 8db1893
Showing 1 changed file with 5 additions and 2 deletions.
7 changes: 5 additions & 2 deletions src/hipscat/io/parquet_metadata.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
"""Utility functions for handling parquet metadata files"""

import tempfile
from typing import List
from typing import Any, Dict, List, Union

import numpy as np
import pyarrow as pa
Expand Down Expand Up @@ -73,7 +73,10 @@ def get_healpix_pixel_from_metadata(


def write_parquet_metadata(
catalog_path: str, order_by_healpix=True, storage_options: dict = None, output_path: str = None
catalog_path: str,
order_by_healpix=True,
output_path: str = None,
storage_options: Union[Dict[Any, Any], None] = None,
):
"""Generate parquet metadata, using the already-partitioned parquet files
for this catalog.
Expand Down

0 comments on commit 8db1893

Please sign in to comment.