fix: yaml validation files in gcs (#977)

* Fix to accept GCS paths in configs run -c * Blacken files
GoogleCloudPlatform · Sep 28, 2023 · bf0fa0a · bf0fa0a
1 parent 8df1cfa
commit bf0fa0a
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/data_validation/cli_tools.py b/data_validation/cli_tools.py
@@ -1063,8 +1063,16 @@ def get_validation(validation_name, config_dir=None):
         mgr = state_manager.StateManager(file_system_root_path=config_dir)
         return mgr.get_validation_config(validation_name, config_dir)
     else:
-        mgr = state_manager.StateManager()
-        return mgr.get_validation_config(validation_name)
+        if validation_name.startswith("gs://"):
+            obj_depth = len(validation_name.split("/"))
+            gcs_prefix = "/".join(validation_name.split("/")[: obj_depth - 1])
+            mgr = state_manager.StateManager(file_system_root_path=gcs_prefix)
+            return mgr.get_validation_config(
+                validation_name.split("/")[obj_depth - 1], gcs_prefix
+            )
+        else:
+            mgr = state_manager.StateManager()
+            return mgr.get_validation_config(validation_name)
 
 
 def list_validations():