From 8db18931d1bb171482cd23c23d55668258f76a5e Mon Sep 17 00:00:00 2001
From: Melissa DeLucchi <113376043+delucchi-cmu@users.noreply.github.com>
Date: Fri, 26 Jul 2024 16:36:18 -0400
Subject: [PATCH] Better mypy hints (#316)

---
 src/hipscat/io/parquet_metadata.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/src/hipscat/io/parquet_metadata.py b/src/hipscat/io/parquet_metadata.py
index 7dd258b5..46617e2f 100644
--- a/src/hipscat/io/parquet_metadata.py
+++ b/src/hipscat/io/parquet_metadata.py
@@ -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
@@ -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.