improved shap baseline uri schema validation

aws-byeldos · aws-byeldos · commit 0a69d86c6b91 · 2022-08-24T12:02:50.000+02:00
diff --git a/src/sagemaker/clarify.py b/src/sagemaker/clarify.py
@@ -88,7 +88,7 @@ def _validate_s3_path(path: str) -> str:
             Optional("shap"): {
                 Optional("baseline"): Or(
                     # URI of the baseline data file
-                    str,
+                    And(str, _validate_s3_path),
                     # Inplace baseline data (a list of something)
                     [
                         Or(

Original file line number	Diff line number	Diff line change
`@@ -88,7 +88,7 @@ def _validate_s3_path(path: str) -> str:`
`88`	`88`	`Optional("shap"): {`
`89`	`89`	`Optional("baseline"): Or(`
`90`	`90`	`# URI of the baseline data file`
`91`		`- str,`
	`91`	`+ And(str, _validate_s3_path),`
`92`	`92`	`# Inplace baseline data (a list of something)`
`93`	`93`	`[`
`94`	`94`	`Or(`