Add optionality

nikhilwoodruff · nikhilwoodruff · commit 7d04966ca398 · 2025-04-29T08:57:45.000-04:00
diff --git a/policyengine/utils/data_download.py b/policyengine/utils/data_download.py
@@ -9,8 +9,8 @@
 class DataFile(BaseModel):
     filepath: str
     huggingface_org: str
-    huggingface_repo: str = None
-    gcs_bucket: str = None
+    huggingface_repo: str | None = None
+    gcs_bucket: str | None = None
 
 
 def download(
