Cloud options are really optional

rapidsai · Jul 22, 2024 · 6aa9432 · 6aa9432
1 parent 31533a1
commit 6aa9432
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/cudf_polars/cudf_polars/dsl/ir.py b/python/cudf_polars/cudf_polars/dsl/ir.py
@@ -210,7 +210,7 @@ def __post_init__(self) -> None:
         if self.typ not in ("csv", "parquet"):
             raise NotImplementedError(f"Unhandled scan type: {self.typ}")
         if self.cloud_options is not None and any(
-            self.cloud_options[k] is not None for k in ("aws", "azure", "gcp")
+            self.cloud_options.get(k) is not None for k in ("aws", "azure", "gcp")
         ):
             raise NotImplementedError(
                 "Read from cloud storage"