Rename Schema Inspection names in the torch schema module #1179

oliverholworthy · 2023-07-04T15:38:37Z

Add schema inspection functions to top-level merlin.models.torch module.

input_schema - schema describing the inputs expected by the module
output_schema - schema describing the outputs of the module
feature_schema - schema describing the expected features passed to the module during training
target_schema - schema describing the expected targets passed to the module during training

Renames schema.input to schema.input_schema, and schema.output to schema.output_schema

Exisitng tests updated to reflect new names

github-actions · 2023-07-04T15:49:34Z

Documentation preview

oliverholworthy added 4 commits July 3, 2023 16:15

Create input_schema and output_schema methods on Block

22a77e3

Expose schema functions from merlin.models.torch

1bed9d9

Rename input/output schema and add to top level torch module

c742dc6

Revert input/output schema methods on Block

937fdfd

oliverholworthy added the enhancement New feature or request label Jul 4, 2023

oliverholworthy added this to the Merlin 23.07 milestone Jul 4, 2023

oliverholworthy self-assigned this Jul 4, 2023

oliverholworthy requested a review from marcromeyn July 4, 2023 15:41

oliverholworthy and others added 2 commits July 4, 2023 16:41

Remove whitespace form torch __init__.py

b03f1b2

Merge branch 'main' into torch-schema-functions

9b3a8d3

marcromeyn approved these changes Jul 4, 2023

View reviewed changes

oliverholworthy and others added 2 commits July 5, 2023 11:34

Merge branch 'main' into torch-schema-functions

d37c5f9

Re-format schema.py

0173138

oliverholworthy merged commit eddc2ab into NVIDIA-Merlin:main Jul 5, 2023
35 checks passed

oliverholworthy deleted the torch-schema-functions branch July 5, 2023 12:43