Add support for google storage (#63)

ray-project · Mar 3, 2021 · 5d65f24 · 5d65f24
1 parent 5caec15
commit 5d65f24
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/xgboost_ray/matrix.py b/xgboost_ray/matrix.py
@@ -69,6 +69,7 @@ def _is_petastorm_compatible(data: Union[str, List[str]]):
 
     def _is_compatible(url: str):
         return url.endswith(".parquet") and (url.startswith("s3://")
+                                             or url.startswith("gs://")
                                              or url.startswith("hdfs://")
                                              or url.startswith("file://"))