gpflow.experimental.check_shapes.parser#

Functions#

gpflow.experimental.check_shapes.parser.parse_and_rewrite_docstring#

gpflow.experimental.check_shapes.parser.parse_and_rewrite_docstring(docstring, function_spec, context)[source]#

Rewrite docstring to include the shapes specified by the argument_specs.

Parameters:
Return type:

Optional[str]

gpflow.experimental.check_shapes.parser.parse_function_spec#

gpflow.experimental.check_shapes.parser.parse_function_spec(function_spec, context)[source]#

Parse all check_shapes argument or note specification for a single function.

Parameters:
  • function_spec (Sequence[str]) –

  • context (ErrorContext) –

Return type:

ParsedFunctionSpec

gpflow.experimental.check_shapes.parser.parse_tensor_spec#

gpflow.experimental.check_shapes.parser.parse_tensor_spec(tensor_spec, context)[source]#

Parse a check_shapes tensor specification.

Parameters:
Return type:

ParsedTensorSpec