Skip to content

Commit 8db1893

Browse files
authored
Better mypy hints (#316)
1 parent 3234fcf commit 8db1893

1 file changed

Lines changed: 5 additions & 2 deletions

File tree

src/hipscat/io/parquet_metadata.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
"""Utility functions for handling parquet metadata files"""
22

33
import tempfile
4-
from typing import List
4+
from typing import Any, Dict, List, Union
55

66
import numpy as np
77
import pyarrow as pa
@@ -73,7 +73,10 @@ def get_healpix_pixel_from_metadata(
7373

7474

7575
def write_parquet_metadata(
76-
catalog_path: str, order_by_healpix=True, storage_options: dict = None, output_path: str = None
76+
catalog_path: str,
77+
order_by_healpix=True,
78+
output_path: str = None,
79+
storage_options: Union[Dict[Any, Any], None] = None,
7780
):
7881
"""Generate parquet metadata, using the already-partitioned parquet files
7982
for this catalog.

0 commit comments

Comments
 (0)