Skip to content

Conversation

@pranavm-nvidia
Copy link
Collaborator

  • Adds tests to ensure we only use ASCII characters in our code and markdown files. Some non-ASCII characters that look like ASCII characters were breaking tests that make assumptions about the encoding.

  • Updates dtype constraint tests to properly construct inputs for various APIs.

  • Moves dtype constraint tests to L0 since they only add about 20 seconds and were breaking very often otherwise.

  • Updates binary elementwise ops to only return ShapeScalars if all inputs are also ShapeScalars.

  • Corrects dtype constraints for several APIs.

  • Simplifies mean implementation.

  • Corrects type annotations for some APIs and introduces a ShapeLike type.

  • Adds support for Optional during type checking.

  • Updates convert_inputs_to_tensors to now potentially convert tensors to Tensor subclasses. See the note in the code for details on the logic.

Copy link
Contributor

@slyubomirsky slyubomirsky left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Some minor comments, though we could leave broader discussion to #233. Thank you for the various corrections and improvements.

- Adds tests to ensure we only use ASCII characters in our code and markdown files.
     Some non-ASCII characters that look like ASCII characters were breaking tests
     that make assumptions about the encoding.

- Updates dtype constraint tests to properly construct inputs for various APIs.

- Moves dtype constraint tests to L0 since they only add about 20 seconds and were
     breaking very often otherwise.

- Updates binary elementwise ops to only return `ShapeScalar`s if *all* inputs are also
     `ShapeScalar`s.

- Corrects dtype constraints for several APIs.

- Simplifies `mean` implementation.

- Corrects type annotations for some APIs and introduces a `ShapeLike` type.

- Adds support for `Optional` during type checking.

- Updates `convert_inputs_to_tensors` to now potentially convert tensors to
     `Tensor` subclasses. See the note in the code for details on the logic.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants