diff --git a/data_validation/cli_tools.py b/data_validation/cli_tools.py index 0b26bdd16..68455f264 100644 --- 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():